提示词

一个以日期为index的DataFrame，包含一列股票的收盘价，现在需要用TA-Lib计算其均线指标并可视化，请给出示例代码。

答案

import pandas as pd
import talib
import matplotlib.pyplot as plt
# import yfinance as yf
# 获取股票数据（示例使用苹果公司2024年数据）
# data = yf.download("AAPL", start="2024-01-01", end="2025-06-23")
# data = data[['Close']].rename(columns={'Close': 'close'})

# Python代码
#准备数据的代码，已经用过多次
#这里不再逐行注释
df = pd.read_excel('../数据/历史行情数据.xlsx')
df = df[['日期', '收盘']]
df['日期'] = pd.to_datetime(df['日期'])
df.set_index('日期', inplace = True)

# 使用TA-Lib计算均线
df['SMA_20'] = talib.SMA(df['收盘'], timeperiod=20)  # 20日简单均线
df['EMA_12'] = talib.EMA(df['收盘'], timeperiod=12)  # 12日指数均线
df['EMA_26'] = talib.EMA(df['收盘'], timeperiod=26)  # 26日指数均线[1,3](@ref)

# 过滤掉前25天的无效数据（EMA26需要至少26个数据点）
valid_data = df.dropna()

plt.figure(figsize=(14, 7))
plt.title('均线指标图', fontsize=14)
plt.xlabel('Date', fontsize=12)
plt.ylabel('Price (USD)', fontsize=12)

# 绘制价格曲线
plt.plot(valid_data['收盘'], label='Close Price', color='#1f77b4', linewidth=1.5)

# 绘制均线系统
plt.plot(valid_data['SMA_20'], label='20-Day SMA', 
         color='orange', linestyle='--', linewidth=1.2)
plt.plot(valid_data['EMA_12'], label='12-Day EMA', 
         color='red', linewidth=1.2)
plt.plot(valid_data['EMA_26'], label='26-Day EMA', 
         color='purple', linewidth=1.2)

# 添加辅助元素
plt.grid(True, alpha=0.3)
plt.legend(loc='upper left', fontsize=10)
plt.tight_layout()
plt.show()

我增加了自己的数据和修改了图标题和表格列名，避免中文乱码的代码，其余都是ai写的。

图像：