python - numpy 中的小计

Question

我有一个 numpy 数组:

[[1 3 1]
 [8 9 0]
 [1 3 1]
 [8 4 1]
 [5 1 0]]

我想为第 3 列的每个值生成第 0,1 列的小计(计数、总和、平均值)。可以直接在 numpy 中完成，还是必须循环整个数组？

Answer 1

如果第三列中的值不多，您可以对每个值都这样做(假设您的数组是 data):

np.mean(data[data[:,2] == 1], axis = 0)
np.sum(data[data[:,2] == 1], axis = 0)

否则，您可以循环查找第三列中的不同值。