python读取tiff文件进行波段计算_在python中利用GDAL对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