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

Question

我有一个像这样的 numpy 数组:

[[1, 2, 3, 4],
 [2, 2, 3, 4],
 [1, 4, 5, 6],
 [3, 2, 3, 4]]

我想要做的是平均每个重复的第一个条目的行。在此示例中，计算第一行和第三行倒数第二列的平均值。正确的输出应该是:

[[1, 3, 4, 5],
 [2, 2, 3, 4],
 [3, 2, 3, 4]]

这个答案Average entries with duplicate first element in 2D numpy array与我想做的事情有很大关系。

Answer 1

你会用 Pandas 吗？

import pandas as pd

df = pd.DataFrame(
    [[1, 2, 3, 4],
     [2, 2, 3, 4],
     [1, 4, 5, 6],
     [3, 2, 3, 4]], 
    columns=list('abcd')
    )

df.groupby('a').mean().reset_index().values
# returns:
array([[1, 3, 4, 5],
       [2, 2, 3, 4],
       [3, 2, 3, 4]], dtype=int64)