将二维numpy数组保存为单波段tif图

import gdal
import numpy as np

shuzu = np.full((5, 5), 3.1) #生成一个5X5大小的矩阵，每一个值都是3.1
print(shuzu)

# 创建tif文件
driver = gdal.GetDriverByName("GTiff")
New_YG_dataset = driver.Create(r'F:\YG\result\综合夜间灯光强度.tif', 5, 5, 1, gdal.GDT_Float32)

# 这里也可以读取之前tif的信息，为新生成的tif添加相关信息，
# 如果不添加这些信息，生成的就是没有坐标信息的图片。
# YG_dataset = gdal.Open(r'F:\YG\data\SVDNB_npp_20181201-20181213caijian.tif')  # 打开文件
# YG_proj = YG_dataset.GetProjection()  # 地图投影信息
# New_YG_dataset.SetProjection(YG_proj)
# YG_geotrans = YG_dataset.GetGeoTransform()  # 仿射矩阵
# New_YG_dataset.SetGeoTransform(YG_geotrans)

band = New_YG_dataset.GetRasterBand(1)
band.WriteArray(shuzu)
print(New_YG_dataset)
New_YG_data = New_YG_dataset.ReadAsArray(0, 0, 3, 3)  # 将数据写成数组，对应栅格矩阵
print(New_YG_data)