1dブール値配列を使用した2d配列

Question

2つのnumpy配列があります。

filtered_x = filter(x,y) 
print(filtered_x) # [[1,2], [5,6]] should be shown. 

私はnp.extractを試みたが、xが1次元配列の場合にのみ動作するようです：私はyの対応する要素がTrueとなっているXの要素を取得したいのですが

x = [[1,2], [3,4], [5,6]] 
y = [True, False, True] 

の要素を抽出するには、yの対応する値はTrueですか？

Answer 1

だけboolean indexingを使用します。

>>> import numpy as np 

>>> x = np.array([[1,2], [3,4], [5,6]]) 
>>> y = np.array([True, False, True]) 
>>> x[y] # or "x[y, :]" because the boolean array is applied to the first dimension (in this case the "rows") 
array([[1, 2], 
     [5, 6]]) 

そして、あなたが代わりに行の列にそれを適用したい場合：

>>> x = np.array([[1,2], [3,4], [5,6]]) 
>>> y = np.array([True, False]) 
>>> x[:, y] # boolean array is applied to the second dimension (in this case the "columns") 
array([[1], 
     [3], 
     [5]]) 

Answer 2

l=[x[i] for i in range(0,len(y)) if y[i]]、これはそれを行います。