from osgeo import gdal_array

# 将源图片载入到数组中
srcArray = gdal_array.LoadFile("TIF/FalseColor/FalseColor.tif")

# print(len(arr))
# >> 3
# print(len(arr[0]), " ", len(arr[1]), " ", len(arr[2]))
# 2693 2693 2693
# 包含3个波长的图片，由长度为3的数组组成.每个波长含2693个数据

# 为了获得一张自然色彩的图片 交换波段1和波段2的位置
# 将使用numpy库的 "高级分片" 功能对波段进行重新排列
# 调用源图片
gdal_array.SaveArray(srcArray[[1, 0, 2], :], "TIF/FalseColor/swap.tif", format="GTiff",
                              prototype="TIF/FalseColor/FalseColor.tif")
# 取消输出避免在某些平台上损坏文件