python - 通过python中的唯一值计算数组组的平均值

Question

如果我有一个数组

[0,0,0,0,0,0,1,1,2,2,2,2]

如何使用数组中的每个唯一值计算平均值numpy.mean()。

我不想[6/12, 2/12,4/12]使用 count 或 len 但只np.mean

我刚从 Python 开始。

标签： pythonarrayspython-3.xnumpygroup-by

Answer 1

如何使用数组中的每个唯一值计算平均值numpy.mean()。

numpy.mean可能无法使用。

但要实现这一点，您可以使用numpy.bincount和len

import numpy as np

np.bincount(a)/len(a)
array([0.5       , 0.16666667, 0.33333333])