Numpy数组多个掩码

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)

有没有办法更快地做到这一点，也就是没有＆＃34; for＆＃34;循环？

Answer 1

使用np.bincount -

的一种方法

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

当mask中的元素不一定是从0开始的顺序时，np.unique的另一个{{3}}用于通用案例 -

_,ids, count = np.unique(mask, return_inverse=1, return_counts=1)
out = np.bincount(ids, data)/count