python - Excel 的 SUMIF 函数的 Python 列表或 NumPy 等价物是什么？

Question

我有一个二维数组:

expenses = np.array([['jim', 'sam', 'bill', 'sam'],[1,2,6,5]])

我想知道新数组中每个唯一人员的总费用而不对任何名称进行硬编码(实际列表很长)，以便我得到如下输出:

totals = [['jim', 'sam', 'bill'],[1,7,6]]

有没有办法用列表或 NumPy 来做到这一点？我不想为此使用 Pandas。

提前致谢!

Answer 1

names = np.asarray(['jim', 'sam', 'bill', 'sam'])
values = np.asarray([1, 2, 6, 5])
result = {name: values[names == name].sum() for name in np.unique(names)}