在python中利用GDAL对tif文件进行读写的方法
利用GDAL库对tif影像进行读取
示例代码默认波段为[B、G、R、NIR的顺序，且为四个波段]
importgdal defreadTif(fileName): dataset=gdal.Open(fileName) ifdataset==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文件函数
importgdal
importnumpyasnp
defwriteTiff(im_data,im_width,im_height,im_bands,im_geotrans,im_proj,path):
if'int8'inim_data.dtype.name:
datatype=gdal.GDT_Byte
elif'int16'inim_data.dtype.name:
datatype=gdal.GDT_UInt16
else:
datatype=gdal.GDT_Float32
iflen(im_data.shape)==3:
im_bands,im_height,im_width=im_data.shape
eliflen(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)#写入投影
foriinrange(im_bands):
dataset.GetRasterBand(i+1).WriteArray(im_data[i])
deldataset
以上这篇在python中利用GDAL对tif文件进行读写的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。