Python 中的 TreeMap - 垂直和水平

Question

我有下面的代码，它只是在 Python 中创建了一个 Treemap。谁能帮助我使用下面的代码示例在 Treemap 或 Horizontal 中仅创建所有垂直矩形。

import matplotlib.pyplot as plt
import squarify
volume = [350, 220, 170, 150, 50]
labels = ['A', 'B', 'C', 'D', 'E']
color_list = ['#0f7216', '#b2790c', '#ffe9a3', '#f9d4d4', '#d35158', '#ea3033']
plt.rc('font', size=14)
squarify.plot(sizes=volume, label=labels, color=color_list, alpha=0.7)
plt.axis('on')
plt.show()

它应该看起来像水平或垂直堆栈条形图。但是在 Python 中使用 Treemap 创建的任何选项。因为我想进一步修改它。下面是建议的外观。

所需的垂直输出

期望水平输出

感谢您的帮助。

Answer 1

您可以使用 norm_x 和 norm_y 指定适合数据的矩形形状（默认为 100 x 100 正方形，因此得名 squarify):

水平：

squarify.plot(sizes=volume, norm_x=100, norm_y=10, label=labels, color=color_list, alpha=0.7)

垂直：

squarify.plot(sizes=volume, norm_x=10, norm_y=100, label=labels, color=color_list, alpha=0.7)

详情见documentation。

Python 中的 TreeMap - 垂直和水平

TreeMap In Python - Vertical and Horizontal

python

graph

matplotlib

treemap