前言

本文的文字及图片来源于网络,仅供学习、交流使用,不具有任何商业用途,如有问题请及时联系我们以作处理。

以下文章来源于可以叫我才哥 ，作者才哥

大家好，上次我们试着用vba在excel中绘制树状热力图，但是我在绘制全部4000+股票的时候等待1小时最终效果图还远没完成，那么咋办呢/p>

终于在最近学习plotly中，让我在高级图表里发现了treemap，居然可以很好地满足我的需求，大家看以下就是最终效果图，是不是很赞！

看这个图吧，我就又要哭了，我重仓了白酒啊！！希望快点涨涨涨吧！！！

目录：

• 1. 准备工作
• 2. 开始绘图
  • 2.1. 简单的例子
  • 2.2. px.treemap常用参数介绍
  • 2.3. color_continuous_scale参数介绍
  • 2.4. 大A股市树状热力图来了
  • 2.5. plotly图片存本地

Python爬虫、数据分析、网站开发等案例教程视频免费在线观看

Python学习交流群：1039649593

1. 准备工作

我这边是在jupyterlab中演示的plotly图表，如果只安装plotly是无法正常显示图表的（会显示为空白），我们需要进行以下准备(以下命令均在cmd下操作即可)：

参考：
https://github.com/plotly/plotly.py

关于本文用到的数据呢，大家可以参考《python爬取股票最新数据并用excel绘制树状图》的爬虫代码自行爬取，或者公众号回复0302获取哈！

2. 开始绘图

Treemap绘制方式有两种，一种是在plotly.express下，一种是在go.Treemap下，我们这里用到的是前者。至于其中的区别，大概就是前者是高级版本，封装了很多后者的复杂操作，可以直接用pandas.Dataframe类型，是现在主推的。更多差异大家可自行在实践过程中感悟，毕竟才哥也说不太明白！

2.1. 简单的例子

plotly自带很多测试数据，我们用其中一个才试试简单的。

以上数据大致就是星期几什么餐不同性别的人支付的金额和小费数等，我们用1行代码绘制简单的treemap如下：

可以看到在该树状图中，层级依次是在path中顺序的day、time和sex，而每个色块面积大小就是values给定的total_bill。鼠标悬停的时候，会出现更详细的信息，毕竟plotly是可交互式的，这里不展开哈。

那么，px.treemap究竟还可以做什么呢们继续往下看。

2.2. px.treemap常用参数介绍

让我们直接在jupyterlab用获取函数参数吧！

对于更多参数的了解，大家可以自己 nbsp;后看哈，蛮详细的，就是纯英文的我自己看着要结合翻译软件也蛮累。

2.3. color_continuous_scale参数介绍

上面我们提到过该参数为 我们 treemap图色块颜色使用的色卡，具体有哪些可选以及这些可选项都是啥样呢/p>

我们在找到了方向：

我们在jupyterlab里调用里面提到的方法，很开心的发现了它们：

由于在我们国家，绿色代表跌，红色代表涨，寻觅了半天我选取了Geyser，当然大家亦可自行选择。

2.4. 大A股市树状热力图来了

基于以上的了解，发现差不多可以绘制图了，那么来吧！

2.5. plotly图片存本地

这里是指图片哦，不是html文件哈！

大家还记得我们在准备工作中 安装orca ，没事，不记得的话，重新装一下就行了。

对于我们绘制好的plotly图表对象，以下方式可以进行本地保存。

当然，要是觉得麻烦，其实你还可以直接点击图表最右上角的摄像头进行下载哈！

文章知识点与官方知识档案匹配，可进一步学习相关知识Python入门技能树首页概览208568 人正在系统学习中

来源：松鼠爱吃饼干