您可以使用bincount（）：
import numpy as npids = [1,1,1,2,2,2,3]data = [20,30,0,4,8,9,18]print np.bincount(ids, weights=data)
输出为[0. 50. 21. 18.]，这意味着id == 0的总和为0，id == 1的总和为50。
您可以使用bincount（）：
import numpy as npids = [1,1,1,2,2,2,3]data = [20,30,0,4,8,9,18]print np.bincount(ids, weights=data)
输出为[0. 50. 21. 18.]，这意味着id == 0的总和为0，id == 1的总和为50。
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
