過濾器，並在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。所以，只要切片，並拿出總和。