Python修改图像指定坐标像素强度的for循环失效如何解决

错误原因 #

你原有代码的判断逻辑存在根本性错误：
brighter_spot.any()的作用是检查坐标数组中是否存在非零元素，仅会返回True或False的布尔值，你将图像的像素强度值与布尔值做相等判断，二者几乎不可能匹配，因此条件永远不会触发，像素修改自然没有效果。
同时你已经明确拿到了所有需要修改的像素坐标，不需要额外做比对判断，直接对对应位置赋值即可。

修复方法 #

方案1：保留for循环的最简修改 #

直接删除错误的判断条件，对遍历到的坐标直接赋值：

for y,x in brighter_spot:
    fig300[0,y,x] = 0

如果确实需要满足某个条件才修改（例如原像素强度高于阈值才设为0），可以修改判断条件为对应阈值比对即可，示例：

# 示例：仅原像素强度大于200时才设为0
threshold = 200
for y,x in brighter_spot:
    if fig300[0,y,x] > threshold:
        fig300[0,y,x] = 0

方案2：更高效的NumPy向量化操作（推荐） #

NumPy数组原生支持批量索引赋值，不需要写for循环，运行效率会高很多，尤其适合坐标量较大的场景：

# 提取所有y、x坐标序列
y_list = brighter_spot[:, 0]
x_list = brighter_spot[:, 1]
# 批量赋值
fig300[0, y_list, x_list] = 0

如果需要避免重复坐标的无效重复操作，可以先对坐标数组去重后再操作：

import numpy as np
# 坐标去重
unique_spots = np.unique(brighter_spot, axis=0)
fig300[0, unique_spots[:,0], unique_spots[:,1]] = 0

内容的提问来源于stack exchange，提问作者Frepp