最近在参加数模比赛,在培训课程中老师要求使用python完成对于K线图制作。参考了网上很多个博客,终于找到适合自己环境的在此记录下。
我使用的环境如下:
操作系统:windows10
编译器:pycharm2019.2
python环境:anaconda2020.07
题目分析:本题中制作K线图的数据老师已经提供,以.csv的形式给出。python制作K线图有专门的库,我们练习的主要目的是可视化呈现。数据的获取不是关键。对于K线图数据获取可使用tushare库函数实现,因为我不需要这步,在此不再陈述(其实是我不会~)
制作K线图代码如下:
import datetime
import csv
import matplotlib.pyplot as plt
import mpl_finance as mpf
import matplotlib.dates as mpd
import numpy as np
str2date = lambda x: mpd.date2num(datetime.datetime.strptime(x, '%Y-%m-%d').date())
#使用numpy库中loadtxt函数从文件中获取数据
data_list = np.loadtxt("000004.csv", str, delimiter=",", skiprows=1) # delimiter代表分隔符,skiprows=1代表跳过第一行
columns = data_list[0]
#对于data_list[1:53]由于老师限定了时间范围,也使用部分数据制作。此处可自由调整
quotes = [[str2date(d[0])] + [float(v) for v in d[1:-1]] for d in data_list[1:53]]
fig, ax = plt.subplots(figsize=(6, 4), facecolor=(0.5, 0.5, 0.5)) # fig画布 ax对象子图
fig.subplots_adjust(bottom=0.2)
mpf.candlestick_ohlc(ax, quotes, width=0.4, colorup='r', colordown='g') # k线图函数
plt.grid(False)
ax.xaxis_date()
ax.autoscale_view()
plt.setp(plt.gca().get_xticklabels(), rotation=30)
plt.title("Candlestick chart")
plt.xlabel("Date")
plt.ylabel("Price")
plt.show()
最终效果图如下:
感谢网上各位大神博客作为参考,因为看了太多,无法一一列举,请谅解!如有不足之处,请下方评论区指出,谢谢!