如何在python中栅格化矢量数据(线)并沿线插入值？

import gdal
import fiona
import numpy as np
from scipy.interpolate import griddata

with fiona.open('input.shp', 'r') as src:
    lines = [feature['geometry']['coordinates'] for feature in src]

xmin, ymin, xmax, ymax = src.bounds

resolution = 10  # 栅格化分辨率
grid_size = int(max(xmax - xmin, ymax - ymin) / resolution)

grid = np.zeros((grid_size, grid_size))

for line in lines:
    # 将线要素的坐标转换为栅格坐标
    x = np.array([int((coord[0] - xmin) / resolution) for coord in line])
    y = np.array([int((coord[1] - ymin) / resolution) for coord in line])
    
    # 对线要素进行插值操作
    values = np.linspace(0, 1, len(line))  # 假设插值值为0到1的线性变化
    interpolated_values = griddata((x, y), values, (np.arange(grid_size), np.arange(grid_size)), method='linear')
    
    # 将插值结果赋值给栅格数组
    grid = np.maximum(grid, interpolated_values)

driver = gdal.GetDriverByName('GTiff')
output = driver.Create('output.tif', grid_size, grid_size, 1, gdal.GDT_Float32)
output.GetRasterBand(1).WriteArray(grid)
output.SetGeoTransform((xmin, resolution, 0, ymin, 0, resolution))
output.FlushCache()
output = None