所以我有一個 numpy 數(shù)組：array([[2. , 0.125], [3. , 0.125], [3. , 0.125], [4. , 0.125], [4. , 0.125], [5. , 0.125], [5. , 0.125], [6. , 0.125]])如果第一個匹配如下，我想將這些條目合并在一起：array([[2. , 0.125], [3. , 0.25 ], [4. , 0.25 ], [5. , 0.25 ], [6. , 0.125]])他們是一種無需循環(huán)即可解決此問題的方法嗎？
2 回答
有只小跳蛙
TA貢獻1824條經(jīng)驗 獲得超8個贊
您可以為此使用numpy.bincount 。
import numpy as np
a = np.array([[2.? ?, 0.125],
? ? ? ? ? ? ? [3.? ?, 0.125],
? ? ? ? ? ? ? [3.? ?, 0.125],
? ? ? ? ? ? ? [4.? ?, 0.125],
? ? ? ? ? ? ? [4.? ?, 0.125],
? ? ? ? ? ? ? [5.? ?, 0.125],
? ? ? ? ? ? ? [6.? ?, 0.125],
? ? ? ? ? ? ? [5.? ?, 0.125]],
)
def groupby(a):
? ? """
? ? >>> groupby(a)
? ? array([[2.? ?, 0.125],
? ? ? ? ? ?[3.? ?, 0.25 ],
? ? ? ? ? ?[4.? ?, 0.25 ],
? ? ? ? ? ?[5.? ?, 0.25 ],
? ? ? ? ? ?[6.? ?, 0.125]])
? ? """
? ? values, indices = np.unique(a[:, 0], return_inverse=True)
? ? sum = np.bincount(indices, weights=a[:, 1])
? ? return np.c_[values, sum]
心有法竹
TA貢獻1866條經(jīng)驗 獲得超5個贊
如果您考慮其他套餐，pandas是一個不錯的選擇：
pd.DataFrame(a).groupby([0])[1].sum()
輸出：
0
2.0 0.125
3.0 0.250
4.0 0.250
5.0 0.250
6.0 0.125
Name: 1, dtype: float64
添加回答
舉報
0/150
提交
取消