在前面的文章中,我们试着用mpl_finance这个库来试着画蜡烛图(详见文章:Python:蜡烛图),但mpl_finance这个库不是太友好,横纵轴的自适应能力不强,需要调整好久,有没有新的库来更好地画出蜡烛图。Now,它来了,plotly就相当可以。我们以平安银行(‘000001.sz’)自2010年以来的股价数据来做一个蜡烛图,数据如图:
数据来源:Tushare
试试plotly的效果吧,具体代码为:
import pandas as pdimport plotly.graph_objects as goimport datetime#导入相关数据df=pd.read_csv('000001.csv')df=df.iloc[::-1]#数据格式转换df['trade_date']=df['trade_date'].apply(lambda x: datetime.datetime.strptime(str(x),'%Y%m%d'))#绘制20日均线df['20wma']=df['close'].rolling(window=20).mean()fig=go.Figure(data=[go.Candlestick(x=df['trade_date'],open=df['open'],high=df['high'] ,low=df['low'],close=df['close'])])fig.add_trace(go.Scatter(x=df['trade_date'],y=df['20wma'],line={'color':'blue'},name='20wma'))fig.update_layout(xaxis_rangeslider_visible=False,template='plotly_dark')fig.show()
运行结果如下:
效果不错,代码简单,plotly不愧为python中的数据可视化神器: