004：画出K线图

我们尝试继续画出K线图，用京泉华的数据，文件是002885.csv。

（数据获取，参考前文《001:如何获取A股个股的前复权K线数据》）

新建文件test004.py

import pandas as pd
import matplotlib.pyplot as plt
from mplfinance.original_flavor import candlestick_ohlc
import matplotlib.dates as mdates

# 读取CSV文件
df = pd.read_csv('002885.csv')

# 将日期列转换为日期类型
df['日期'] = pd.to_datetime(df['日期'])

# 创建一个新的列，包含日期的序列号
df['日期序号'] = df['日期'].apply(mdates.date2num)

# 创建一个子图
fig, ax = plt.subplots()

# 绘制K线图
candlestick_ohlc(ax, df[['日期序号', '开盘', '最高', '最低', '收盘']].values, width=0.6, colorup='g', colordown='r')

# 设置x轴的刻度格式
ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m-%d'))

# 设置x轴的刻度间隔
ax.xaxis.set_major_locator(mdates.DayLocator())

# 设置图表标题
plt.title('candle chat')

# 显示图表
plt.show()

执行效果如下：