Boolean Индексация нП Numpy Массив

Question

Я не мог применить значения BOOL в приведенных данных:

import numpy as np 

bool_values = np.array([False, False, True, False, True]) 
data = np.array([[11.0, 22.0, 0.0, 44.0, 55.0], 
       [10.0, 20.0, 30.0, 40.0, 50.0]]) 

expected_answer = [[0,55], [30,50]] 

Я попытался как:

expected_answer = data[bool_values] 

Answer 1

Я думаю, что вы хотели бы сделать это:

In [28]: 
bool_values = np.array([False, False, True, False, True]) 
data = np.array([[11.0, 22.0, 0.0, 44.0, 55.0], 
[10.0, 20.0, 30.0, 40.0, 50.0]]) 
data[:,bool_values] 

Out[28]: 
array([[ 0., 55.], 
     [ 30., 50.]]) 

Вышеприведенная шкала применяется к каждой строке