如何使用rasterio修改TIFF影像的波段数（从3改为4）

直接修改元数据中的count值会导致TIFF文件损坏，因为TIFF的波段数据是实际存储的，必须同步添加对应的波段数据。以下是正确的实现方案：

核心逻辑 #

读取原有影像的所有波段数据，添加新的波段数据后重新写入文件，同时更新元数据的count字段为4。

代码示例 #

1. 新增全填充值的波段（以全0为例） #

import rasterio
import numpy as np

# 读取原影像
with rasterio.open('input.tiff', 'r') as src:
    # 复制原影像元数据并修改波段数量
    meta = src.meta.copy()
    meta['count'] = 4
    
    # 读取原有3个波段数据
    existing_bands = src.read()
    # 创建新波段（尺寸与原影像一致，填充0）
    new_band = np.zeros((src.height, src.width), dtype=meta['dtype'])
    
    # 合并原有波段与新波段
    all_bands = np.concatenate([existing_bands, new_band[np.newaxis, :, :]])
    
    # 写入新文件（若需覆盖原文件，建议先备份）
    with rasterio.open('output.tiff', 'w', **meta) as dst:
        dst.write(all_bands)

2. 使用自定义数据作为新波段 #

如果新波段有实际业务数据，只需替换new_band为你的数据即可，注意数据类型要与原影像保持一致（比如原影像为uint8，新波段也需是uint8）：

import rasterio
import numpy as np

# 假设已准备好新波段数据，尺寸与原影像匹配（499*524）
custom_data = np.random.randint(0, 255, size=(499, 524), dtype=np.uint8)

with rasterio.open('input.tiff', 'r') as src:
    meta = src.meta.copy()
    meta['count'] = 4
    existing_bands = src.read()
    # 合并原有波段与自定义新波段
    all_bands = np.concatenate([existing_bands, custom_data[np.newaxis, :, :]])
    
    with rasterio.open('output.tiff', 'w', **meta) as dst:
        dst.write(all_bands)

关键注意事项 #

• 禁止直接通过src.count = 4修改原文件元数据，这会破坏文件结构，导致影像无法正常读取。
• 若要覆盖原文件，务必先备份原始数据，避免数据丢失。
• 新波段的尺寸（高度、宽度）必须与原影像完全一致，数据类型需匹配原影像的dtype参数。

内容的提问来源于stack exchange，提问作者Salvatore Danilo Palumbo