STA/LTA算法代码

STA/LTA（Short-term Average / Long-term Average）算法是一种用于检测信号中事件的方法，常用于地震学和其他领域。该算法通过计算信号的短期平均值（STA）和长期平均值（LTA），并将它们进行比较以确定事件的发生。当短期平均值显著超过长期平均值时，通常意味着一个事件被检测到。

下面是一个简单的STA/LTA算法的Python代码示例：

import numpy as np

def sta_lta(signal, sta_length, lta_length):
    """
    计算STA/LTA比率的函数。
    :param signal: 输入的一维信号数组
    :param sta_length: 短期平均窗口长度
    :param lta_length: 长期平均窗口长度
    :return: STA/LTA比率数组
    """
    sta = np.cumsum(signal ** 2)
    sta[sta_length:] = sta[sta_length:] - sta[:-sta_length]
    sta /= sta_length
    lta = np.cumsum(signal ** 2)
    lta[lta_length:] = lta[lta_length:] - lta[:-lta_length]
    lta /= lta_length
    
    return sta / lta

# 示例使用
# 假设data是我们需要处理的信号数组
data = np.random.normal(size=1000)
# 设置短期平均窗口长度为50，长期平均窗口长度为150
ratio = sta_lta(data, sta_length=50, lta_length=150)

# ratio数组中大于1的点表示潜在的事件发生点

这个函数首先计算了信号的短期和长期平方累计和，然后分别除以它们的窗口长度得到平均值，并最终计算短期平均值除以长期平均值的比率。