Python绘制K线图代码
时间: 2023-11-05 17:59:34 浏览: 106
可以这样实现:
```python
import matplotlib.pyplot as plt
from mpl_finance import candlestick_ohlc
import pandas as pd
import numpy as np
df = pd.read_csv("path/to/your/csv/file.csv")
# 设置画布大小
fig = plt.figure(figsize=(10, 6))
# 添加子图
ax = fig.add_subplot(1, 1, 1)
# 修改 x 轴为日期格式
df["date"] = np.arange(start=0, stop=len(df))
df["date"] = pd.to_datetime(df["date"], unit="D")
df = df.set_index("date", drop=False)
# 转换为 OHLC 格式
ohlc = []
for date, row in df.iterrows():
open_price, high_price, low_price, close_price = row[["Open", "High", "Low", "Close"]]
ohlc.append([date.date(), open_price, high_price, low_price, close_price])
# 绘制 K 线图
candlestick_ohlc(ax, ohlc, width=0.6, colorup="green", colordown="red")
# 设置 x 轴为日期格式
ax.set_xlabel("Date")
ax.xaxis.set_major_formatter(plt.DateFormatter("%Y-%m-%d"))
fig.autofmt_xdate()
# 设置 y 轴为股价
ax.set_ylabel("Price")
# 设置标题
plt.title("Stock K-Line")
# 显示图形
plt.show()
```
注意,以上代码只是一个简单的示例,具体实现需要根据你的数据结构和需求进行修改。同时,本 AI 不对代码的正确性负责,仅供参考。
阅读全文