Python中基于坐标列表修改图像像素为白色的TypeError问题排查

• 遍历逻辑错误：你用for i in x_fit时，i是x_fit里的浮点数值，不是列表索引，接着用x_fit[i]相当于用浮点数作为列表下标，直接触发TypeError。
• 坐标配对错误：x_fit和y_fit是一一对应的坐标对（比如x_fit[0]对应y_fit[0]），嵌套循环会把所有x和y的组合遍历一遍，完全不符合画直线的逻辑。
• 像素坐标类型错误：回归计算得到的坐标是浮点型，而图像数组的索引必须是整数。
• 坐标顺序错误：OpenCV图像数组的索引是[行, 列]，对应图像的y, x坐标，你之前写的reg_img[x_fit[i], y_fit[y]]顺序完全反了，这也是潜在问题。

方案1：修正遍历逻辑，转换坐标为整数 #

直接配对x_fit和y_fit的对应元素，转成整数后修改像素，同时处理边界避免越界：

# 在(x_fit, y_fit)位置绘制白色像素
for x, y in zip(x_fit, y_fit):
    # 转换为整数索引，round或int按需选择
    x_idx = int(round(x))
    y_idx = int(round(y))
    # 确保坐标在图像范围内，防止越界报错
    if 0 <= x_idx < reg_img.shape[1] and 0 <= y_idx < reg_img.shape[0]:
        reg_img[y_idx, x_idx] = (255, 255, 255)
cv2.imshow('reg_img', reg_img)

方案2：用OpenCV直接画直线（更高效） #

如果x_fit是覆盖图像宽度的连续x值，直接取首尾坐标画直线即可，无需逐个像素修改：

# 取首尾坐标并转换为整数
start_point = (int(round(x_fit[0])), int(round(y_fit[0])))
end_point = (int(round(x_fit[-1])), int(round(y_fit[-1])))
# 绘制白色直线，thickness参数可调整线条粗细
cv2.line(reg_img, start_point, end_point, (255, 255, 255), thickness=1)
cv2.imshow('reg_img', reg_img)

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