根据标准差拉伸波段创建新栅格(.tif)，适用于数据堆栈，但不能写入新文件Python

import gdal
import numpy as np

# 打开原始栅格数据
dataset = gdal.Open('input.tif')

# 读取波段数据
band = dataset.GetRasterBand(1)
band_data = band.ReadAsArray()

# 计算标准差
std_dev = np.std(band_data)

# 计算拉伸范围
min_value = np.mean(band_data) - 2 * std_dev
max_value = np.mean(band_data) + 2 * std_dev

# 根据标准差拉伸波段
stretched_band_data = (band_data - min_value) / (max_value - min_value) * 255

# 创建新的栅格数据
driver = gdal.GetDriverByName('GTiff')
output_dataset = driver.Create('output.tif', dataset.RasterXSize, dataset.RasterYSize, 1, gdal.GDT_Byte)
output_dataset.SetProjection(dataset.GetProjection())
output_dataset.SetGeoTransform(dataset.GetGeoTransform())

# 写入拉伸后的波段数据
output_band = output_dataset.GetRasterBand(1)
output_band.WriteArray(stretched_band_data)

# 释放资源
dataset = None
output_dataset = None