def plot_stock_average(stock_data, name):
stock_data['trade_date'] = pd.to_datetime(stock_data['trade_date'], format='%Y%m%d')
stock_data.set_index('trade_date', inplace=True)
stock_data[name.title()] = stock_data[name]
stock_data[name.title()].plot()
stock_data['SMA5'] = stock_data[name].rolling(window=5).mean()
stock_data['SMA5'].plot()
stock_data['EMA5'] = stock_data[name].ewm(span=5).mean()
stock_data['EMA5'].plot()
plt.xlabel('Date')
plt.ylabel('Price')
plt.title(f'{name.title()} Stock Prices')
plt.legend()
plt.grid(True)
plt.show()
画出股票均线
最新推荐文章于 2025-06-04 16:07:42 发布