一行 Python 代码轻松构建树状热力图

一行 Python 代码轻松构建树状热力图

今天和大家一起学习一种可视化技术:构建树状热力图treemap。树形图易于可视化,且易于被人理解。树状图通过展示不同大小的矩形,以传达不同大小的数据量,一般认为,较大的矩形意味着占总体的一大部分,而较小的矩形意味着整体的一小部分。在本文中,云朵君将和大家一起学习如何使用Squarify库在 Python 中构建树形图。

介绍

树状图使用嵌套在一起的不同大小的矩形来可视化分层数据。每个矩形的大小与其代表的整体数据量成正比。这些嵌套的矩形代表树的分支,因此得名。除了尺寸外,每个矩形都有代表独特类别的独特颜色。树状图广泛用于从金融机构到销售组织的行业。

树形图最早是在 1990 年代早期由美国马里兰大学人机交互实验室的 Ben Shneiderman 教授发明的。这种可视化背后的想法是在固定空间中按大小比较数量。现在,我们将看看如何实际构建词云。

一行 Python 代码轻松构建树状热力图

佛罗里达州的树状图 来源:https://commons.wikimedia.org

使用 Squarify 构建树状图

Python 中的,可以使用 Squarify 直接构建树状图。且只需要一行代码squarify.plot(data)即可轻松构建。

1. 安装必要的库

!pip install squarify

2. 导入必要的库

import matplotlib.pyplot as plt
import squarify

3. 创建随机数据

随机生成一个值列表,这些值将作为数据传递到我们的绘图中。

data = [500, 250, 60, 120]

4. 绘制树状图

使用**squarify.plot()** 方法构建树状图。这里将随机数据变量data作为此 squarify.plot 方法的参数。此外,附加一行修饰代码 plt.axis('off') 隐藏 树状图 的轴。

squarify.plot(data)
plt.axis('off')
plt.show()

5. 把它们放在一起

import matplotlib.pyplot as plt
import squarify
data = [500, 250, 60, 120]
squarify.plot(data)
plt.axis('off')
plt.show()

在执行此代码时,我们得到:

一行 Python 代码轻松构建树状热力图

每次执行此代码时,它都生成一组随机颜色的嵌套矩形。

使用附加参数

借助 .plot() 方法的参数,可以在树状图中添加更多的修饰。可以通过明确指定属性来控制树形图的颜色、标签和填充。

1. 指定树状图的颜色

import matplotlib.pyplot as plt
import squarify
sizes = [500, 250, 120, 60]
color = ['red', 'green', 'blue', 'orange']
squarify.plot(sizes,color=color)
plt.axis('off')
plt.show()

在执行此代码时,我们得到:

一行 Python 代码轻松构建树状热力图

2.给树状图添加标签

不同的标签值可以通过将列表传递到 squarify.plot() 的标签属性来显式添加标签。这将覆盖现有标签或将标签添加到我们的树状图中(如果不存在)。标签将按照.plot()所传递的列表中的相同顺序被添加到树状图中。

import matplotlib.pyplot as plt
import squarify
labels = ['A', 'AB', 'ABC', 'ABCD']
sizes = [500, 250, 120, 60]
color = ['red', 'green', 'blue', 'orange']
squarify.plot(sizes,color=color, label = labels)
plt.axis('off')
plt.show()

在执行此代码时,我们得到:

一行 Python 代码轻松构建树状热力图

3. 树状图中的pad

可以在树状图中添加pad,将树状图中的每个具体彼此分离,这将有助于更好地区分矩形。当有大量类别或矩形时,这很有用。可以通过将pad参数设置为True来调用。

import matplotlib.pyplot as plt
import squarify
labels = ['AB', 'A', 'ABC', 'ABCD']
sizes = [500, 250, 120, 60]
color = ['red', 'green', 'blue', 'orange']
squarify.plot(sizes,color=color, label = labels, pad = True)
plt.axis('off')
plt.show()

在执行代码时,我们得到:

一行 Python 代码轻松构建树状热力图

写在最后

由此看来,构建树形图可谓是小菜一碟。除了squarify 库,树状图还可以使用 Python 中的其他几个库来构建。如比较流行的plotly库。在今天的次条推文中介绍了其应用案例,感兴趣的小伙伴可以看看。

import plotly_express as px
import plotly.graph_objects as go
px.treemap(names = name, parents = parent)
go.Figure(go.Treemap(labels = name, parents = parent,))

当然,还有许多 BI 工具可用于更加方便简单地构建树状图。

有时,树状图中可能会出现歧义。如果有多个具有相同数量(或矩形大小)和相同颜色深浅的类别,则导致用户最终很难区分它们。所以在构建树状图时,必须始终考虑所涉及的类别数量和颜色映射。

以上就是一行 Python 代码轻松构建树状热力图的详细内容,更多请关注https://www.sxiaw.com/其它相关文章!