如何按指定交替规则实现500×500×3 numpy空白图像的像素标记

实现思路 #

首先明确两个基础前提：

• 尺寸为500×500×3的图像总共有250000个独立像素，常规图像处理默认按行优先顺序遍历像素：即从图像左上角开始，从左到右走完第一行再走第二行，直到右下角最后一个像素，每个像素对应一个从0开始的连续整数索引。
• 给定的Mark序列按顺序交替对应「跳过的像素长度」「需要填充标记的像素长度」：序列第1个值（索引0）是初始跳过长度，第2个值（索引1）是第一段填充长度，第3个值（索引2）是下一段跳过长度，以此类推循环。

直接操作numpy数组的切片比逐像素循环计算(x,y)坐标效率高得多，不需要单独换算每个像素的行列位置，处理完连续索引后数组会自动对应回原图像的行列位置。

可直接运行的代码 #

import numpy as np

# 初始化空白图像
Img = np.zeros((500, 500, 3), dtype=np.uint8)
# 替换为你实际使用的Mark序列
Mark = [2, 455, 6, 556, 12, 654, 22, 23, 4, 86]
# 替换为你需要填充的标记色，三个值分别对应三个通道，比如纯白色为(255,255,255)
fill_color = (255, 255, 255)

total_pixel = 500 * 500
current_ptr = 0
# 将三维图像展平为[像素序号, 通道值]的二维数组，展平后的数组是原数组的视图，修改会直接同步到原图像
flat_img = Img.reshape(-1, 3)

for seq_idx, step in enumerate(Mark):
    if current_ptr >= total_pixel:
        break
    # 自动截断超出图像总像素范围的步长，避免索引报错
    valid_step = min(step, total_pixel - current_ptr)
    if seq_idx % 2 == 0:
        # 偶数位为跳过长度，直接移动指针
        current_ptr += valid_step
    else:
        # 奇数位为填充长度，给对应切片的像素统一赋值标记色
        flat_img[current_ptr : current_ptr + valid_step] = fill_color
        current_ptr += valid_step

# 处理完成后Img数组就是标记完成的图像，可直接用于后续保存、显示操作

补充说明 #

• 如果需要手动换算某个连续索引对应的(x,y)坐标，直接用整除和取余即可：y = 索引 // 500，x = 索引 % 500，其中y是行号、x是列号，原点在图像左上角。
• 如果Mark序列的步长总和小于250000，未覆盖到的像素会保留初始的黑色值；如果步长总和超过250000，代码会自动在图像末尾截断，不会出现越界错误。
• 填充色的通道顺序要和你后续使用的图像库匹配：用OpenCV读写显示时通道顺序是BGR，用PIL、matplotlib读写显示时通道顺序是RGB，对应调整三个通道的数值顺序即可。

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