标准化降水指数python代码

以下是计算标准化降水指数（SPI）的 Python 代码示例：

import numpy as np
import scipy.stats as st

def spi(data, timescale):
    """
    计算标准化降水指数（SPI）
    :param data: 降雨量序列，一维数组
    :param timescale: 时间尺度，单位为月，例如3表示计算3个月的SPI值
    :return: SPI序列，一维数组
    """
    # 计算降雨量累积值
    cumsum = np.cumsum(data)
    
    # 计算平均值和标准差
    window = st.norm.ppf((cumsum[timescale-1:] - cumsum[:-timescale+1])/timescale)
    mean = np.mean(window)
    std = np.std(window)
    
    # 计算SPI值
    spi = (window[-1] - mean) / std
    
    return spi

使用时，可以将每个时间段内的降雨量数据传入 `spi()` 函数中计算对应的 SPI 值，例如：

data = [50, 70, 80, 60, 40, 30, 20, 10, 50, 60, 70, 80]
timescale = 3
spi_values = spi(data, timescale)
print(spi_values)

其中 `data` 列表中存储的是每个时间段内的降雨量数据，`timescale` 参数表示计算 SPI 值的时间尺度。运行结果将输出 SPI 值序列。