用Python写个读取股票数据绘制k线图
时间: 2024-09-25 08:04:04 浏览: 51
在Python中,你可以使用一些库如`pandas`处理金融数据,如股票价格,然后结合`matplotlib`或`plotly`来绘制K线图。以下是一个简单的示例,假设我们已经从网络上获取了股票数据并存储在一个CSV文件中:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取股票数据(这里假设数据列有日期、开盘价、收盘价、最高价、最低价)
df_stock = pd.read_csv('stock_data.csv', parse_dates=['date'])
# 定义函数绘制K线图
def plot_candlestick(df):
fig, ax = plt.subplots()
ax.xaxis_date() # 设置x轴为日期格式
df['open_close'] = df[['open', 'close']].values.tolist() # 创建一个新列用于绘图
ax.plot(df['date'], df['open_close'], color='black', linestyle='-', linewidth=0.5)
for i in range(1, len(df)):
open_, close_ = df.iloc[i - 1]['open_close']
high, low = df.iloc[i]['high'], df.iloc[i]['low']
ax.bar(df['date'][i], (close_ - open_) / 2, bottom=open_, width=1, color='green' if close_ > open_ else 'red')
ax.set_ylabel('Price')
plt.title('Stock K-line Chart')
plt.show()
# 调用函数绘制K线图
plot_candlestick(df_stock)
阅读全文