在 Python 中，可以通过 GDAL 库将地理转换 (`geotransform`) 和投影 (`projection`) 信息写入输出影像文件中。以下是详细的实现方式： ### 使用 GDAL 创建带有 `geotransform` 和 `projection` 的 TIFF 文件 #### 1. 导入库 首先导入必要的库来操作影像数据。 ```python from osgeo import gdal, osr import numpy as np ``` #### 2. 定义地理变换参数 地理变换矩阵是一个长度为 6 的列表或元组，表示如下： - 左上角 X 坐标 (`gt[0]`) - 像素宽度 (`gt[1]`) - 图像旋转项（通常为 0）(`gt[2]`) - 左上角 Y 坐标 (`gt[3]`) - 图像旋转项（通常为 0）(`gt[4]`) - 像素高度（负数表示向下增加）(`gt[5]`） ```python # 地理变换参数示例 geotransform = [444720, 30, 0, 3751320, 0, -30] ``` #### 3. 设置投影信息 定义目标投影系统，并将其导出为 WKT 格式以便存储到文件中。 ```python # 创建空间参考对象 srs = osr.SpatialReference() # 配置投影信息（例如 UTM Zone 11N） srs.SetUTM(11, True) srs.SetWellKnownGeogCS("WGS84") # 转换为 WKT 格式的字符串 projection = srs.ExportToWkt() ``` #### 4. 创建输出数据集 使用 GDAL 创建新的 GeoTIFF 数据集，并设置其大小、波段数量和像素类型。 ```python # 打开驱动程序 driver = gdal.GetDriverByName('GTiff') # 输出文件路径 output_file = 'output_with_georef.tif' # 影像尺寸 rows, cols = 512, 512 # 波段数量和数据类型 bands = 1 dtype = gdal.GDT_Float32 # 创建新影像 dst_ds = driver.Create(output_file, cols, rows, bands, dtype) if dst_ds is None: raise RuntimeError("Failed to create output file.") ``` #### 5. 添加地理变换和投影信息 将前面定义的 `geotransform` 和 `projection` 写入数据集中。 ```python # 设置地理变换 dst_ds.SetGeoTransform(geotransform) # 设置投影信息 dst_ds.SetProjection(projection) ``` #### 6. 写入影像数据 准备一个 NumPy 数组作为测试数据填充至新建的 GeoTIFF 文件当中。 ```python # 测试数据：随机生成浮点型数值 data_array = np.random.rand(rows, cols).astype(np.float32) # 将数组写入第一个波段 band = dst_ds.GetRasterBand(1) band.WriteArray(data_array) # 可选：设置无数据值 nodata_value = -9999 band.SetNoDataValue(nodata_value) ``` #### 7. 清理资源 完成所有操作后，确保释放资源以保存更改。 ```python # 刷新缓存并关闭数据集 dst_ds.FlushCache() del dst_ds ``` --- ### 总结 上述代码展示了如何利用 GDAL 库创建一个包含地理变换和投影信息的 GeoTIFF 文件。其中的关键步骤包括配置 `geotransform` 参数、定义投影系统以及将这些信息绑定到输出数据集中[^1]。