如何过滤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]])