文章目录
大家好,上次我们试着用 vba在excel中绘制树状热力图,但是我在绘制全部4000+股票的时候等待1小时最终效果图还远没完成,那么咋办呢?
终于在最近学习plotly中,让我在高级图表里发现了treemap,居然可以很好地满足我的需求,大家看以下就是最终效果图,是不是很赞!

看这个图吧,我就又要哭了,我重仓了白酒啊!!希望快点涨涨涨吧!!!
1. 准备工作
我这边是在jupyterlab中演示的plotly图表,如果只安装plotly是无法正常显示图表的(会显示为空白),我们需要进行以下准备(以下命令均在cmd下操作即可):
# 安装plotly库及plotly-orca库
pip install plotly
pip install plotly plotly-orca
# Basic JupyterLab renderer support
jupyter labextension install jupyterlab-plotly
# OPTIONAL: Jupyter widgets extension for FigureWidget support
jupyter labextension install @jupyter-widgets/jupyterlab-manager plotlywidget
参考:https://github.com/plotly/plotly.py
关于本文用到的数据呢,大家可以参考《python爬取股票最新数据并用excel绘制树状图》的爬虫代码自行爬取,或者公众号回复0302获取哈!
2. 开始绘图
Treemap绘制方式有两种,一种是在plotly.express下,一种是在go.Treemap下,我们这里用到的是前者。至于其中的区别,大概就是前者是高级版本,封装了很多后者的复杂操作,可以直接用pandas.Dataframe类型,是现在主推的。更多差异大家可自行在实践过程中感悟,毕竟才哥也说不太明白!
2.1. 简单的例子
plotly自带很多测试数据,我们用其中一个才试试简单的。
import plotly.express as px
df = px.data.tips()
df.head()

以上数据大致就是星期几什么餐不同性别的人支付的金额和小非数等,我们用1行代码绘制简单的treemap如下:
fig = px.treemap(df, path=['day', 'time', 'sex'], values='total_bill')
fig.show



被折叠的 条评论
为什么被折叠?



