多毫安阵列

Question

尝试基于整数掩码数组对numpy数组进行多次切片和平均：

即

import numpy as np

data = np.arange(11)
mask = np.array([0, 1, 1, 1, 0, 2, 2, 3, 3, 3, 3])

results = list()
for maskid in range(1,4):
    result = np.average(data[mask==maskid])
    results.append(result)
output = np.array(result)

有没有一种方法可以更快地做到这一点，也就是说没有“for”循环？在

Answer 1

一种使用^{}-

np.bincount(mask, data)/np.bincount(mask)

另一个带有^{}的例子，当mask中的元素不一定是从0开始的序列时-

^{pr2}$