如何在GDAL中创建栅格堆栈并将TIFF文件名设置为波段名称？

当然可以实现这个需求！GDAL的BuildVRT和Translate工具默认不会自动将文件名设为波段名称，但我们可以在生成堆栈后手动给每个波段设置描述（也就是软件中显示的波段名）。下面是修改后的完整代码和步骤说明：

完整代码示例 #

import glob
import os
from osgeo import gdal

# 定义输出路径
outvrt = 'result/raster_stack_vrt.tif'
outtif = 'result/raster_stack.tif'
# 获取所有要堆叠的TIFF文件
tifs = glob.glob('data/*.tif')

# 第一步：创建VRT并转换为TIFF
outds = gdal.BuildVRT(outvrt, tifs, separate=True)
outds = gdal.Translate(outtif, outds)
# 必须关闭数据集，确保文件写入磁盘
outds = None

# 第二步：打开生成的TIFF，设置波段名称
ds = gdal.Open(outtif, gdal.GA_Update)
if ds is None:
    print(f"Error: 无法打开文件 {outtif}")
else:
    # 遍历每个文件和对应的波段
    for idx, tif_path in enumerate(tifs):
        # 提取纯文件名（去掉路径和.tif后缀）
        band_name = os.path.splitext(os.path.basename(tif_path))[0]
        # GDAL波段索引从1开始，所以要+1
        band = ds.GetRasterBand(idx + 1)
        # 设置波段描述，这就是软件中显示的波段名称
        band.SetDescription(band_name)
    # 保存修改并关闭数据集
    ds = None

print("波段名称设置完成！")

关键步骤说明 #

• 关闭数据集：在生成TIFF后必须执行outds = None，因为GDAL采用延迟写入机制，不关闭的话文件可能仍在内存中，后续无法正常修改。
• 提取纯文件名：os.path.splitext(os.path.basename(tif_path))[0]会把类似data/soil_type_2019.tif的路径转换成soil_type_2019，刚好作为清晰的波段名。
• 可写模式打开：用gdal.GA_Update模式打开TIFF，才能对文件进行修改操作。
• 波段索引：GDAL中波段的索引是从1开始的，所以循环时要把列表的0索引对应到波段1。

这样设置后，不管你用QGIS、ArcGIS还是其他支持GDAL的GIS软件打开这个栅格堆栈，每个波段都会显示对应的原始文件名，再也不会混淆啦！

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