我有一个二维数组:
expenses = np.array([['jim', 'sam', 'bill', 'sam'],[1,2,6,5]])
我想知道新数组中每个唯一人员的总费用而不对任何名称进行硬编码(实际列表很长),以便我得到如下输出:
totals = [['jim', 'sam', 'bill'],[1,7,6]]
有没有办法用列表或 NumPy 来做到这一点?我不想为此使用 Pandas。
提前致谢!
最佳答案
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)}
关于python - Excel 的 SUMIF 函数的 Python 列表或 NumPy 等价物是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51642321/