Python中基于GDAL库的TIF影像读写实现指南

一只小鹿

发布时间：2026-10-13 | 5人浏览过

原创

本文详细介绍如何在Python环境中利用GDAL库实现TIF影像文件的读取与写入操作。内容涵盖获取影像尺寸、波段数及地理变换参数，提取B、G、R、NIR四个波段数据，以及根据数据类型动态创建TIF文件并写入多波段数据的完整代码示例，为遥感数据处理提供基础技术支持。

通过调用GDAL库，可以实现对TIF格式影像文件的有效读取操作。

在示例代码中，假设影像包含四个波段，且波段排列顺序依次为蓝色（B）、绿色（G）、红色（R）和近红外（NIR）。

import gdal
def readTif(fileName):
 dataset = gdal.Open(fileName)
 if dataset == None:
  print(fileName+"文件无法打开")
  return
 im_width = dataset.RasterXSize #栅格矩阵的列数
 im_height = dataset.RasterYSize #栅格矩阵的行数
 im_bands = dataset.RasterCount #波段数
 im_data = dataset.ReadAsArray(0,0,im_width,im_height)#获取数据
 im_geotrans = dataset.GetGeoTransform()#获取仿射矩阵信息
 im_proj = dataset.GetProjection()#获取投影信息
 im_blueBand = im_data[0,0:im_height,0:im_width]#获取蓝波段
 im_greenBand = im_data[1,0:im_height,0:im_width]#获取绿波段
 im_redBand = im_data[2,0:im_height,0:im_width]#获取红波段
 im_nirBand = im_data[3,0:im_height,0:im_width]#获取近红外波段

以下是用于将处理后的数据写入TIF影像文件的函数实现。

#保存tif文件函数
import gdal
import numpy as np
def writeTiff(im_data,im_width,im_height,im_bands,im_geotrans,im_proj,path):
 if 'int8' in im_data.dtype.name:
  datatype = gdal.GDT_Byte
 elif 'int16' in im_data.dtype.name:
  datatype = gdal.GDT_UInt16
 else:
  datatype = gdal.GDT_Float32

 if len(im_data.shape) == 3:
  im_bands, im_height, im_width = im_data.shape
 elif len(im_data.shape) == 2:
  im_data = np.array([im_data])
 else:
  im_bands, (im_height, im_width) = 1,im_data.shape
  #创建文件
 driver = gdal.GetDriverByName("GTiff")
 dataset = driver.Create(path, im_width, im_height, im_bands, datatype)
 if(dataset!= None):
  dataset.SetGeoTransform(im_geotrans) #写入仿射变换参数
  dataset.SetProjection(im_proj) #写入投影
 for i in range(im_bands):
  dataset.GetRasterBand(i+1).WriteArray(im_data[i])
 del dataset

以上就是在Python中利用GDAL库对TIF文件进行读写操作的完整方法分享，希望能为读者提供有价值的参考，同时也欢迎大家继续支持本站获取更多技术内容。