python - 二维 numpy 数组中具有重复第一个元素的平均条目

Question

我有一个看起来像这样的数组

  arr = np.array([[0, 1], [0, 2], [1, 3], [1, 3], [1, 4], [2, 3]])

我想取具有相同第一个元素的“条目”的平均值，即我的输出应该是

  [ [0, avg(1,2)] , [1, avg(3, 3, 4)], [2, 3] ]

执行此操作的最佳方法是什么？

Answer 1

这是一个使用 np.unique 的 NumPythonic 解决方案和 np.bincount对于第一列并不总是排序的一般情况 -

unqa,ID,counts = np.unique(arr[:,0],return_inverse=True,return_counts=True)
out = np.column_stack(( unqa , np.bincount(ID,arr[:,1])/counts ))

sample 运行-

In [4]: arr
Out[4]: 
array([[5, 1],
       [5, 2],
       [1, 3],
       [1, 3],
       [5, 4],
       [2, 3]])

In [5]: unqa,ID,counts = np.unique(arr[:,0],return_inverse=True,return_counts=True)
   ...: out = np.column_stack(( unqa , np.bincount(ID,arr[:,1])/counts ))
   ...: 

In [6]: out
Out[6]: 
array([[ 1.        ,  3.        ],
       [ 2.        ,  3.        ],
       [ 5.        ,  2.33333333]])