用python选择一只股票,绘制其过去一年的价格走势图,加上其他两条线:分别代表五日和十日均 线,并在图中绘出基于均线的买入和卖出信号。将图片输出为“1.2”。
时间: 2024-05-07 17:20:28 浏览: 126
好的,首先需要安装一些必要的库,如下所示:
```python
!pip install pandas matplotlib mplfinance
```
然后,我们可以使用以下代码绘制股票价格走势图、5日均线和10日均线,并在图中标注买入和卖出信号。
```python
import pandas as pd
import mplfinance as mpf
import matplotlib.pyplot as plt
# 读取股票数据
df = pd.read_csv('股票数据.csv', index_col=0, parse_dates=True)
# 计算5日均线和10日均线
df['MA5'] = df['Close'].rolling(window=5).mean()
df['MA10'] = df['Close'].rolling(window=10).mean()
# 绘制K线图和5日均线、10日均线
mpf.plot(df, type='candle', mav=(5, 10))
# 标注买入和卖出信号
plt.plot(df[df['Signal'] == 'Buy'].index, df[df['Signal'] == 'Buy']['MA5'], '^', markersize=10, color='g')
plt.plot(df[df['Signal'] == 'Sell'].index, df[df['Signal'] == 'Sell']['MA5'], 'v', markersize=10, color='r')
# 输出图片
plt.savefig('1.2.png')
```
其中,股票数据需要保存在“股票数据.csv”文件中,包含“Date”、“Open”、“High”、“Low”、“Close”五列数据。买入和卖出信号可以通过一些策略计算得出,并保存在“Signal”列中。
最后,该代码将会生成一张名为“1.2.png”的图片,其中包含股票价格走势图、5日均线、10日均线以及买入和卖出信号。
阅读全文