python生成k线图

python生成k线图

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飘逸Desktop600837.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 _xticks()])

# 设置轴标签

ax.set_xlabel('Date', fontsize = 15)

ax.set_ylabel('Price', fontsize = 15)

ax.set_title('600837')

plt.show()