python中怎么利用GDAL讀寫tif文件

這期內容當中小編將會給大家帶來有關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文件了，如果剛好有類似的疑惑，不妨參照上述分析進行理解。如果想知道更多相關知識，歡迎關注億速云行業資訊頻道。