过滤和求和 numpy 数组中的不同列

Question

我有一个大型 numpy 数组 data，我希望按一列过滤 [:,8] <= radius 并获得不同列的总和 [:,7]

到目前为止，我有以下返回“无效切片”错误。

>>> data.slice
(4700, 9)

>>> np.sum(data[np.where(data[:,8] <= 50):,7])
IndexError: invalid slice

我对 python 很陌生，所以似乎真的无法弄清楚我在这里做错了什么。任何想法或解释将不胜感激。

Answer 1

不需要np.where 电话。

data = np.random.normal(size=(20, 2))
np.sum(data[data[:,0] < 0, 1])

在这个例子中，我想要data[:,0] < 0 是True 的行，我想要列1。因此，只需将它们切片并取总和即可。