Python快速过滤Numpy数组中特定元素的方法

最新推荐文章于 2024-07-30 11:56:09 发布

KFXW

最新推荐文章于 2024-07-30 11:56:09 发布

阅读量4.6w

点赞数 10

分类专栏： Python tricks 文章标签： python 数组

本文链接：https://blog.csdn.net/u010158659/article/details/48575179

版权

本文介绍了如何使用Numpy在Python中高效地过滤数组元素，特别是在处理灰度图像数组时，如何通过逻辑运算快速设置数值小于特定阈值的元素为0。这种方法适用于数组筛选和增强可视化效果。

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

对于一个Numpy数组，有的时候我们想选取或者剔除某些符合条件的元素值，却突然发现除了一些特定的函数，如numpy.minimum()等，或者一个嵌套循环之外束手无策。其实这里有一个trick1，可以使代码有效而简洁。

让我们来到一个场景中：我们有一个名为Saliency的灰度图像数组，用来表示一张图片中受关注程度的不同。为了增强可视化效果，我们需要过滤掉数组中数值过小的元素，让其等于0，于是我们可以这么做：

>>> print saliency.shape
(227,227)
>>> idx = saliency < 0.0001  # or some other threshold value
>>> saliency[idx] = 0
# other operations ...

简单

关注

专栏目录