在树图中显示长方体的大小(squarify)

Question

我正在使用python库squarify来绘制树状图：

基本的情节对于我的用例来说很好：

import matplotlib.pyplot as plt 
import squarify  
import pandas as pd 

df = pd.DataFrame({'nb_people':[8,3,4,2], 'group':["group A", "group B", "group C", "group D"] }) 
squarify.plot(sizes=df['nb_people'], label=df['group'], alpha=.8 ) 
plt.axis('off') 
plt.show() 

​

​

但是我想加上盒子的大小和名字。有这样的选择吗？

Answer 1

您可以只在squarify.plot之上添加fig.set_size_inches