python GDAL 影像處理系列之將兩幅3通道影像合併爲一幅6通道影像（四）

以最常見的3通道影像爲例，假設要合併2幅3通道遙感影像爲一幅6通道影像，並且不損失地理信息，就必須要請出我們的主角"GDAL"。

要使用GDAL'的功能，首先要安裝GDAL，具體的安裝教程在GDAL安裝教程。

安裝好了之後就可以簡答使用了，比如gdal的讀寫功能，裁剪拼接功能，，，可以自己小試牛刀一下！

python GDAL 影像處理系列之裁剪與合併影像

python GDAL 影像處理系列之給遙感影像加地理信息

二話不說，直接上代碼：

import gdal
import os
import numpy as np
#讀取tif文件函數
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]#獲取近紅外波段
    im_dtype = im_data.dtype.name
    return im_data, im_width,im_height, im_bands, im_geotrans ,im_proj,im_dtype, im_blueBand, im_greenBand, im_redBand

#保存tif文件函數

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

def get_file_names(data_dir, file_type = ['tif','tiff']):
    result_dir = [] 
    result_name = []
    for maindir, subdir, file_name_list in os.walk(data_dir):
        for filename in file_name_list:
            apath = maindir+'/'+filename
            ext = apath.split('.')[-1]  
            if ext in file_type:
                result_dir.append(apath)
                result_name.append(filename)
            else:
                pass
    return result_dir, result_name



in_dir1 = 'F:/project/cut256_1'
in_dir2 = 'F:/project/cut256_2'
out_dir = 'F:/project/combine256_6bands'

file_type = 'tif'
data_dir_list1,_ = get_file_names(in_dir1, file_type)
data_dir_list2,_ = get_file_names(in_dir2, file_type)
#data_dir_list = data_dir_list1 + data_dir_list2

for each_index, each_dir in enumerate(data_dir_list1):
    img1, width1, height1, bands1, geotrans1, proj1,dtype1, blueband1, greenband1, redband1 = readTif(each_dir)
    img2, width2, height2, bands2, geotrans2, proj2,dtype2, blueband2, greenband2, redband2 = readTif(data_dir_list2[each_index])
    #print(each_dir)
    #print(dtype1)
    if 'int8' in dtype1:
        datatype = gdal.GDT_Byte      
    elif 'int16' in dtype1:
        datatype = gdal.GDT_UInt16
    else:
        datatype = gdal.GDT_Float32
        
            
    driver = gdal.GetDriverByName("GTiff")
   # print(type(driver))
    each_out_dir = out_dir + '/cut256_201903_201906_' + each_dir.split('_')[-4] + '_' + each_dir.split('_')[-3] + '_' + each_dir.split('_')[-2] + '_' + each_dir.split('_')[-1]
    #each_out_dir = 'C:/Users/Dell/Desktop/guigang/trans_6bands.tif'
    #datatype = 'uint8'
    print('each_out_dir: ', each_out_dir)
    new_dataset = driver.Create(each_out_dir, width1, height1, bands1+bands2, gdal.GDT_Byte)
    print(type(new_dataset))
    #print(each_out_dir)
    new_dataset.SetGeoTransform(geotrans1)
    new_dataset.SetProjection(proj1)
    
    new_dataset.GetRasterBand(1).WriteArray(blueband1)
    new_dataset.GetRasterBand(2).WriteArray(greenband1)
    new_dataset.GetRasterBand(3).WriteArray(redband1)
    new_dataset.GetRasterBand(4).WriteArray(blueband2)
    new_dataset.GetRasterBand(5).WriteArray(greenband2)
    new_dataset.GetRasterBand(6).WriteArray(redband2)
    new_dataset = None
    print('combine over')

假如你的數據不是3通道的，直接在讀取完圖像之後獲取每個波段的數據，然後在合併的時候，就可以直接在“new_dataset.GetRasterBand(6).WriteArray(redband2)”後面添加你自己的數據格式就ok了。也可以不像我這樣rgbr'g'b'排列，你也可以選擇rr'gg'bb'等格式，這個看你需求了！

在代碼調試過程中問題，在下方留言就ok

整理不易，謝謝點贊！