在python中使用gdal从csv文件生成tiff文件

import csv
from osgeo import gdal, ogr
from osgeo.gdalconst import *

csv_file = 'path/to/csv/file.csv'
tiff_file = 'path/to/tiff/file.tif'

data = []
with open(csv_file, 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        data.append(row)

driver = gdal.GetDriverByName('GTiff')
dataset = driver.Create(tiff_file, len(data[0]), len(data), 1, GDT_Float32)

for i in range(len(data)):
    for j in range(len(data[i])):
        dataset.GetRasterBand(1).WriteArray(float(data[i][j]), j, i)

dataset.SetGeoTransform((0, 1, 0, 0, 0, 1))
srs = osr.SpatialReference()
srs.ImportFromEPSG(4326)  # 设置地理坐标系为WGS84
dataset.SetProjection(srs.ExportToWkt())

dataset.FlushCache()
dataset = None