python - 通过添加相应的值在 2D numpy 数组中创建唯一行

Question

我在 python 中有一个二维 numpy 数组:

[[ 1  2  1  3  3]
 [10 20 30 40 60]]

我希望在删除列之前第一行中具有唯一值并在第二行中添加相应的值。因此，我的数组的输出将是这样的:

[[  1   2   3 ]
 [ 40  20 100 ]]

我是 python 的新手，我想不出在更大范围内执行此操作的有效方法。

Answer 1

不幸的是，numpy没有内置的groupby函数(尽管有 ways to write them )。如果您愿意使用 pandas，这将很简单:

import pandas as pd

>>> pd.DataFrame(a.T).groupby(0,as_index=False).sum().values.T

array([[  1,   2,   3],
       [ 40,  20, 100]])