【numpy】获取数组满足条件的部分

数组过滤

当使用布尔数组作为下标存取数组中的元素时，将收集数组中所有在数组中对应下标为True的元素，复制创建一个新数组。使用布尔数组作为下标获得的数组不和原始数组共享数据空间，这种方式只对应于布尔数组(array)，不能使用布尔列表（list）。

arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
arr[arr % 2 == 1]
print(arr)

#输出：
# > array([ 0, 2, 4, 6, 8])

取值符中的会进行bool值转换；

arr % 2 == 1

#输出：
#array([False, True, False, True, False, True, False, True, False, True])

所以实际上会转换成用布尔数组作为下标；

arr[np.array([False, True, False, True, False, True, False,  True, False, True])]
#输出：
# array([ 0, 2, 4, 6, 8])

np.where

np.where(condition, x, y)

满足条件(condition)，输出x，不满足输出y。

np.where(condition)

只有条件 (condition)，没有x和y，则输出满足条件 (即非0) 元素的坐标 。这里的坐标以tuple的形式给出，通常原数组有多少维，输出的tuple中就包含几个数组，分别对应符合条件元素的各维坐标。

官网链接