我又用Python爬取了4000+股票数据,并用plotly绘制了树状热力图(treemap)


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

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

3月2日大A头部公司股价又绿晕了啊

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

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()

px.data.tips()

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

fig = px.treemap(df, path=['day', 'time', 'sex'], values='total_bill')
fig.show
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值