如何使用python从值矩阵正确写入栅格图像(tif

import numpy as np
from osgeo import gdal, osr

values = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

rows, cols = values.shape
origin_x = 0  # 图像左上角的x坐标
origin_y = 0  # 图像左上角的y坐标
pixel_width = 1  # 每个像素的宽度
pixel_height = 1  # 每个像素的高度

driver = gdal.GetDriverByName('GTiff')
dataset = driver.Create('output.tif', cols, rows, 1, gdal.GDT_Float32)

srs = osr.SpatialReference()
srs.ImportFromEPSG(4326)  # 设置坐标系，这里使用WGS84经纬度坐标系
dataset.SetProjection(srs.ExportToWkt())
dataset.SetGeoTransform((origin_x, pixel_width, 0, origin_y, 0, pixel_height))

band = dataset.GetRasterBand(1)
band.WriteArray(values)
band.FlushCache()

dataset = None