如何過濾numpy ndarray中的列

Question

我有一些bools數組，表示我應該刪除哪些數據數組列。

selections = [True, False, True] 
data = [[ 1, 2, 3 ], 
     [ 4, 5, 6 ]] 

所以我想有這樣一句：

new_data = [[ 1, 3 ], 
      [ 4, 6 ] 

所有的數組是numpy.array。 Python 2.7版

Answer 1

一旦你實際使用numpy.array S，它的所有作品：

import numpy as np 

selections = np.array([True, False, True]) 
data = np.array([[ 1, 2, 3 ], 
     [ 4, 5, 6 ]]) 

>>> data[:, selections] 
array([[1, 3], 
     [4, 6]])