将上面获取数据的CSV文件中的日期设为索引的代码,并用数据画出MA20,MA50,MA200
时间: 2024-02-25 20:57:29 浏览: 109
用日期为给文件命名的代码
好的,以下是将日期设为索引的代码,并画出MA20,MA50,MA200的代码:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取数据
df = pd.read_csv('data.csv', parse_dates=['date'])
# 将日期设为索引
df.set_index('date', inplace=True)
# 计算MA20,MA50,MA200
df['MA20'] = df['close'].rolling(window=20).mean()
df['MA50'] = df['close'].rolling(window=50).mean()
df['MA200'] = df['close'].rolling(window=200).mean()
# 画图
plt.figure(figsize=(10, 6))
plt.plot(df['close'], label='Close')
plt.plot(df['MA20'], label='MA20')
plt.plot(df['MA50'], label='MA50')
plt.plot(df['MA200'], label='MA200')
plt.legend()
plt.show()
```
注意:需要先安装`pandas`和`matplotlib`库。
阅读全文