在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
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文件进行读写的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。