Python將多個單波段tif合併爲一個影像

• 將同一文件夾裏的多個tif合併爲一個tif或dat或其他的影像文件
• 輸入影像的行列數投影應保持一致
• dat格式可以給波段命名，tif格式看不到波段名稱

# -*- coding:utf-8 -*-
import numpy as np
import gdal
import os
import sys
reload(sys)
sys.setdefaultencoding('utf8')

#修改路徑
tifDir = r"D:\data\testmerge"  #tif路徑 單波段
outtif = r"D:\data\testmerge\result\out1.dat"


NP2GDAL_CONVERSION = {
  "uint8": 1,
  "int8": 1,
  "uint16": 2,
  "int16": 3,
  "uint32": 4,
  "int32": 5,
  "float32": 6,
  "float64": 7,
  "complex64": 10,
  "complex128": 11,
}
tifs = [i for i in os.listdir(tifDir) if i.endswith(".tif")]
#獲取投影波段數等信息
bandsNum = len(tifs)
dataset = gdal.Open(os.path.join(tifDir,tifs[0]))
projinfo = dataset.GetProjection()
geotransform = dataset.GetGeoTransform()
cols,rows=dataset.RasterXSize,dataset.RasterYSize
datatype=dataset.GetRasterBand(1).ReadAsArray(0,0,1,1).dtype.name
gdaltype=NP2GDAL_CONVERSION[datatype]
dataset=None
#創建目標文件
format = "GTiff" #tif格式
format = "ENVI"  # ENVI格式
driver = gdal.GetDriverByName(format)
dst_ds = driver.Create(outtif,cols, rows,bandsNum, gdaltype)
dst_ds.SetGeoTransform(geotransform)
dst_ds.SetProjection(projinfo)
#寫入文件
info = set()
for k in range(bandsNum):
    ds = gdal.Open(os.path.join(tifDir,tifs[k]))
    X,Y = ds.RasterXSize,ds.RasterYSize
    info.add("%s,%s"%(X,Y))
    if(len(info) != 1):
        dst_ds = None
        ds = None
        print("%s 列數行數不一樣：%s,%s"%(tifs[k],X,Y))
        raise Exception("有影像行列數不一致")
    data = ds.GetRasterBand(1).ReadAsArray()    ##讀取第一波段
    ds = None
    dst_ds.GetRasterBand(k+1).WriteArray(data)
    dst_ds.GetRasterBand(k+1).SetDescription("hahha_%s"%k)
    print("波段 %s ==> 文件 %s"%(k+1,tifs[k]))
dst_ds = None