Python中的TreeMap -垂直和水平

Question

下面有代码，它简单地用Python创建树状图。有人可以帮助我只创建树状图或水平的所有垂直矩形使用下面的代码示例。

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来指定矩形的形状以适应数据的需要(默认为100x100平方，因此名为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)

有关细节，请参阅文档。