【numpy】获取数组满足条件的部分
数组过滤
当使用布尔数组作为下标存取数组中的元素时,将收集数组中所有在数组中对应下标为True的元素,复制创建一个新数组。使用布尔数组作为下标获得的数组不和原始数组共享数据空间,这种方式只对应于布尔数组(array),不能使用布尔列表(list)。
arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
arr[arr % 2 == 1] = -1
print(arr)
#输出:
# > array([ 0, -1, 2, -1, 4, -1, 6,
原创
2022-05-02 22:54:45 ·
1412 阅读 ·
0 评论