如何使用numpy中的特定規則從數組中獲取值

Question

例如，我有一個值爲[1,2,4,3,6,7,33,2]的數組。我想要得到大於6的所有值。因爲我知道numpy.take只能得到具有索引的值。
我應該使用哪個函數？

Answer 1

可以索引用布爾數組的索引：

>>> a = np.array([1,2,4,3,6,7,33,2]) 
>>> a > 6 
array([False, False, False, False, False, True, True, False], dtype=bool) 
>>> a[a > 6] 
array([ 7, 33]) 

如果你想去的地方發生這種情況的指數，你可以使用np.where：

>>> np.where(a>6) 
(array([5, 6]),)