mpl_finance是python中可以用来画出蜡烛图、线图的分析工具
安装
pip install mplfinance --upgrade --user -i https://pypi.tuna.tsinghua.edu.cn/simple
单个股票K线图配置颜色
import numpy as np import pandas as pd import tushare as ts import mplfinance as mpf import matplotlib.pyplot as plt from pylab import mpl from datetime import datetime pd.set_option('expand_frame_repr', False)#True就是可以换行显示。设置成False的时候不允许换行 pd.set_option('display.max_columns', None)# 显示所有列 pd.set_option('colheader_justify', 'centre')# 显示居中 pro = ts.pro_api('你的token') mpl.rcParams['axes.unicode_minus'] = False # 解决保存图像是负号'-'显示为方块的问题 df = pro.daily(ts_code='000001.SZ', start_date='20200301', end_date='20201001') #df.sort_values(by='trade_date',ascending=False) data = df.loc[:, ['trade_date', 'open', 'close', 'high', 'low', 'vol']] #:取所有行数据,后面取date列,open列等数据 data = data.rename(columns={'trade_date': 'Date', 'open': 'Open', 'close': 'Close', 'high': 'High', 'low': 'Low', 'vol': 'Volume'}) #更换列名,为后面函数变量做准备 data.set_index('Date', inplace=True) #设置date列为索引,覆盖原来索引,这个时候索引还是 object 类型,就是字符串类型。 data.index = pd.DatetimeIndex(data.index) #将object类型转化成 DateIndex 类型,pd.DatetimeIndex 是把某一列进行转换,同时把该列的数据设置为索引 index。 data = data.sort_index(ascending=True) my_color = mpf.make_marketcolors(up='green', down='red', edge='i', wick='i', volume='in') my_style = mpf.make_mpf_style(marketcolors=my_color, gridaxis='both', gridstyle='-.', y_on_right=False) mpf.plot(data, type='candle', mav=(5, 10, 20),style=my_style, volume=True, show_nontrading=False)
原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/20462.html