[英]Boolean Indexing n-D Numpy Array
我無法將布爾值應用於給定數據:
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]
我認為您想這樣做:
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.]])
上面將遮罩應用於每一行
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.