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

Czyyy_

已于 2022-05-04 16:56:35 修改

阅读量1.4k

点赞数

分类专栏： numpy 文章标签： python

于 2022-05-02 22:54:45 首次发布

本文链接：https://blog.csdn.net/Czyyy_/article/details/124548227

版权

1 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何利用布尔数组在numpy中过滤数组元素，创建新的数组副本而不共享内存。布尔索引能够根据条件选择数组中的特定元素，例如`arr[arr%2==1]`选取奇数。此外，`np.where`函数可以根据条件返回满足条件的元素位置。这些技术在数据处理和分析中非常实用。

摘要由CSDN通过智能技术生成

数组过滤

当使用布尔数组作为下标存取数组中的元素时，将收集数组中所有在数组中对应下标为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(condition, x, y)

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

np.where(condition)

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