如何將一個布爾數組轉換爲索引數組，numpy

有沒有一種有效的Numpy機制來檢索基於條件的數組中的位置的整數索引是真實的，而不是布爾掩碼數組？如何將一個布爾數組轉換爲索引數組，numpy

例如：

x=np.array([range(100,1,-1)]) 
#generate a mask to find all values that are a power of 2 
mask=x&(x-1)==0 
#This will tell me those values 
print x[mask]

在這種情況下，我想知道i的mask其中mask[i]==True索引。有沒有可能產生這些循環？

來源

2011-11-21 Rich

另一種選擇：

In [13]: numpy.where(mask) 
Out[13]: (array([36, 68, 84, 92, 96, 98]),)

這是同樣的事情numpy.where(mask==True)。

來源

2011-11-21 20:49:05

或者類似的，如果你總是有一維數組：'numpy.flatnonzero（mask）'' – christianbrodbeck

您應該可以使用numpy.nonzero()來查找此信息。

來源

2011-11-21 20:41:23 aganders3

使用numpy.nonzero（） [0] 否則你會得到兩個數組。一個帶有索引，另一個帶有值。如果你想使用索引繼續，這更容易。它只給你一個指數數組。 –

np.arange(100,1,-1) 
array([100, 99, 98, 97, 96, 95, 94, 93, 92, 91, 90, 89, 88, 
     87, 86, 85, 84, 83, 82, 81, 80, 79, 78, 77, 76, 75, 
     74, 73, 72, 71, 70, 69, 68, 67, 66, 65, 64, 63, 62, 
     61, 60, 59, 58, 57, 56, 55, 54, 53, 52, 51, 50, 49, 
     48, 47, 46, 45, 44, 43, 42, 41, 40, 39, 38, 37, 36, 
     35, 34, 33, 32, 31, 30, 29, 28, 27, 26, 25, 24, 23, 
     22, 21, 20, 19, 18, 17, 16, 15, 14, 13, 12, 11, 10, 
     9, 8, 7, 6, 5, 4, 3, 2]) 

x=np.arange(100,1,-1) 

np.where(x&(x-1) == 0) 
(array([36, 68, 84, 92, 96, 98]),)

現在重組此類似：

x[x&(x-1) == 0]

來源

2014-06-25 14:39:44 user2290820

如果你喜歡索引方式，您可以將您的布爾列表轉換爲numpy的數組：

print x[nd.array(mask)]

來源

2016-07-29 18:41:40 Serhiy

如何將一個布爾數組轉換爲索引數組，numpy

回答

相關問題