过滤器，并在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。所以，只要切片，并拿出总和。