のpython ...私は、この入力を有効にする関数や方程式を作成する必要が

Question

マスクこの出力に

a = [True, False, True] 
b = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 

...

c = [[1, 3], [4, 6], [7, 9]] 

注意をboolean型の列から配列のサブセットを作成します。上記の配列はすべてnumpy配列です。ありがとう！

Answer 1

numpyソリューション

import numpy as np 
a = np.array([True, False, True]) 
b = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) 
b[:,a] 

array([[1, 3], 
     [4, 6], 
     [7, 9]]) 

Answer 2

あなたは

from itertools import compress 
c = [list(compress(i, a)) for i in b] 

[[1, 3], [4, 6], [7, 9]]