Numpy 布尔索引

import numpy as np
temp = np.array([[1,2,3,4],[5,6,7,8]])
print(temp)

print(temp>2)  # 得到包含 true 和 false 的二维数组
print(temp[temp>2])  # 此处的索引值是二维布尔数组

print("----")
print(temp)

# 手动创建一个 bool数组
bool_arr = np.array([[False,False,True,True],[True,True,True,False]])
print(bool_arr)
print(temp[bool_arr])

[[1 2 3 4]
 [5 6 7 8]]
[[False False  True  True]
 [ True  True  True  True]]
[3 4 5 6 7 8]
----
[[1 2 3 4]
 [5 6 7 8]]
[[False False  True  True]
 [ True  True  True False]]
[3 4 5 6 7]