使用Python生成股票K线图

可视化股票数据，这里只做简单的处理，只显示k线图。选取的是海通证券（600837）2020年1月1日之后150个交易日的数据。这里代码不多，没有封装成方法，代码如下。数据是提前获取的，获取方法见：使用Python获取股票数据 。

import mpl_finance as mpf
from matplotlib import pyplot as plt
import numpy as np
import pandas as pd


fig = plt.figure(figsize=(12,8))
ax = fig.add_subplot(111)
# 加载数据
df_ma = pd.DataFrame()
original_data = pd.read_csv(r'C:\Users\飘逸\Desktop\600837.csv')
print(original_data)
# 设置显示的交易日数
days = 150
# 生成开盘价、收盘价、最高价、最低价
opens = original_data['Open'][0:days]
closes = original_data['Close'][0:days]
highs = original_data['High'][0:days]
lows = original_data['Low'][0:days]
# 生成时间序列
data_index = original_data['Date'][0:days]
# 使用zip()生成数据列表
ohlc = list(zip(np.arange(0,len(opens)), opens, closes, highs, lows))
# 绘制k线图
mpf.candlestick2_ochl(ax, opens, closes, highs, lows, width=0.6, colorup='r', colordown='g')
# 设置x轴的范围
ax.set_xlim(0,days)
# x轴刻度设置
ax.set_xticks(np.arange(0,days,20))
# 标签设置为日期
ax.set_xticklabels([data_index[index] for index in ax.get_xticks()])
# 设置轴标签
ax.set_xlabel('Date', fontsize = 15)
ax.set_ylabel('Price', fontsize = 15)
ax.set_title('600837')
plt.show()

 

显示如下。