別の行列を使用してNumPy行列をサブセット化する

Question

numpy行列がX = np.matrix([[1, 2], [3, 4], [5, 6]])とy = np.matrix([1, 1, 0])であり、y行列に基づいて2つの新しい行列X_posとX_negを作成したいとします。ですから、私の出力はX_pos == matrix([[1, 2], [3, 4]])とX_neg == matrix([[5, 6]])のようになります。これどうやってするの？

Answer 1

yのブール値マスクを作成したい場合は、これが簡単になります。ルーチンnp.ma.masked_whereで

mask = np.array(y).astype(bool).reshape(-1,) 
X_pos = X[mask, :] 
X_neg = X[~mask, :] 

print(X_pos) 
matrix([[1, 2], 
     [3, 4]]) 

print(X_neg) 
matrix([[5, 6]]) 

Answer 2

：

x = np.matrix([[1, 2], [3, 4], [5, 6]]) 
y = np.array([1, 1, 0]) 

m = np.ma.masked_where(y > 0, y) # mask for the values greater than 0 
x_pos = x[m.mask]     # applying masking 
x_neg = x[~m.mask]     # negation of the initial mask 

print(x_pos) 
print(x_neg)