如何遍历rasterio读取的图像像素？解决‘DatasetReader’不可下标错误

你遇到的问题根源是：rasterio.open()返回的DatasetReader对象不是可直接下标的numpy数组，必须先将图像数据读取为numpy数组才能进行像素级访问。

正确步骤与代码示例 #

1. 读取图像为numpy数组

首先用read()方法将DatasetReader中的数据读取为numpy数组（支持masked array），同时用with语句确保文件自动关闭：

import rasterio
import numpy as np

with rasterio.open("FalseColorImages.jpg", masked=True) as img:
    profile = img.profile
    # 读取所有波段，返回形状为 (波段数, 高度, 宽度) 的数组
    img_array = img.read()

2. 遍历像素的几种实现方式

方式一：嵌套循环遍历单波段像素

如果是单波段图像，提取对应波段后直接用行列索引访问：

# 提取第一个波段（单波段图像可直接用img_array）
single_band = img_array[0]
height, width = single_band.shape

for i in range(height):
    for j in range(width):
        pixel_val = single_band[i, j]
        print(pixel_val)

方式二：遍历多波段像素的所有波段值

针对多波段图像，通过[:, i, j]获取对应位置的所有波段值：

bands, height, width = img_array.shape

for i in range(height):
    for j in range(width):
        # 获取(i,j)位置所有波段的像素值，形状为(波段数,)
        pixel_vals = img_array[:, i, j]
        print(pixel_vals)

方式三：numpy高效遍历（适合大图像）

使用np.nditer可以更高效地遍历数组，避免嵌套循环的冗余：

# 遍历单波段每个像素
for pixel in np.nditer(single_band):
    print(pixel)

# 遍历多波段每个像素的完整波段值
for pixel in np.nditer(img_array.T, flags=['external_loop'], order='F'):
    print(pixel)

补充说明 #

• 若图像体积过大，一次性读取全量数据会占用大量内存，可使用Rasterio的window参数分块读取处理。
• masked=True会让读取的数组自动保留掩码信息，无效像素会被标记为掩码状态。

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