Python如何替换NumPy数组中大于某个值的所有元素
发布时间:2020-07-18 09:46:03
来源:亿速云
阅读:84
作者:小猪
这篇文章主要讲解了Python如何替换NumPy数组中大于某个值的所有元素,内容清晰明了,对此有兴趣的小伙伴可以学习一下,相信大家阅读完之后会有帮助。
我有一个2D(二维) NumPy数组,并希望用255.0替换大于或等于阈值T的所有值。据我所知,最基础的方法是:
shape = arr.shape
result = np.zeros(shape)
for x in range(0, shape[0]):
for y in range(0, shape[1]):
if arr[x, y] >= T:
result[x, y] = 255
有更简洁和pythonic的方式来做到这一点吗?
有没有更快(可能不那么简洁和/或不那么pythonic)的方式来做到这一点?
这将成为人体头部MRI扫描窗口/等级调整子程序的一部分,2D numpy数组是图像像素数据。
最佳解决思路
我认为最快和最简洁的方法是使用Numpy的内置索引。如果您有名为arr的ndarray,则可以按如下所示将所有元素>255替换为值xÿ