如何在批量SMA乘法器上绘制买入和做空信号

import pandas as pd
import matplotlib.pyplot as plt

# 示例数据
data = {
    'Date': pd.date_range(start='1/1/2020', periods=100),
    'Close': [i + 10 * (i % 10) for i in range(100)]  # 示例收盘价
}
df = pd.DataFrame(data)

# 计算SMA
df['SMA_5'] = df['Close'].rolling(window=5).mean()
df['SMA_20'] = df['Close'].rolling(window=20).mean()

# 绘制图表
plt.figure(figsize=(14, 7))
plt.plot(df['Date'], df['Close'], label='Close Price')
plt.plot(df['Date'], df['SMA_5'], label='5-day SMA')
plt.plot(df['Date'], df['SMA_20'], label='20-day SMA')

# 标记买入和做空信号
buy_signals = df[df['SMA_5'] > df['SMA_20']]
sell_signals = df[df['SMA_5'] < df['SMA_20']]

plt.scatter(buy_signals['Date'], buy_signals['Close'], color='green', marker='^', label='Buy Signal')
plt.scatter(sell_signals['Date'], sell_signals['Close'], color='red', marker='v', label='Sell Signal')

plt.xlabel('Date')
plt.ylabel('Price')
plt.title('SMA Buy and Sell Signals')
plt.legend()
plt.show()