可视化股票数据,这里只做简单的处理,只显示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()
显示如下。