移动平均

def solve():
    import matplotlib
    matplotlib.use('Agg')
    import matplotlib.pyplot as plt
    import numpy as np
    np.random.seed(42)
    y = np.cumsum(np.random.randn(100))
    window = 10
    ma = np.convolve(y, np.ones(window)/window, mode='valid')
    plt.figure(figsize=(10, 4))
    plt.plot(y, label='Original')
    plt.plot(range(window-1, len(y)), ma, label=f'MA({window})')
    plt.legend()
    plt.title('Moving Average')
    plt.tight_layout()
    plt.show()
    print(len(ma))