如何使用python从x，y，z导出到geotiff

import numpy as np
from osgeo import gdal, osr

x = np.array([x1, x2, x3, ...])
y = np.array([y1, y2, y3, ...])
z = np.array([z1, z2, z3, ...])

projection = 'EPSG:4326'  # 使用WGS84投影坐标系
geotransform = (xmin, xres, 0, ymax, 0, -yres)  # xmin、ymin为左上角坐标，xres、yres为像素分辨率

driver = gdal.GetDriverByName('GTiff')
dataset = driver.Create('output.tif', x.size, y.size, 1, gdal.GDT_Float32)
dataset.SetGeoTransform(geotransform)
dataset.SetProjection(projection)
band = dataset.GetRasterBand(1)

band.WriteArray(z.reshape(y.size, x.size))
band.FlushCache()

dataset = None