是否动态设置Python Squarify图的标签大小?

时间:2019-06-20 21:52:17

标签: python pandas matplotlib squarify

我正在绘制一个带有多个不同参数的树图以进行数据分析。随着盒子变小,文本大小会混杂在一起,变得难以阅读。

我将Squarify库与matplotlib一起使用。

def plot_treemap(df, std):
    title = "Parameter Defect Count - Std +- {0}".format(std)
    std_count = "Def_Count_Std_{0}".format(std)
    norm = matplotlib.colors.Normalize(vmin=df[[std_count]].min(),vmax=df[[std_count]].max())
    colors = [matplotlib.cm.Reds(norm(value)) for value in df["Def_Count_Std_{0}".format(std)]]

    fig = plt.gcf()
    fig.set_size_inches(20, 10)

    squarify.plot(label=df.index.str.replace('_', '\n'), sizes=df["Def_Count_Std_{0}".format(std)], color = colors, alpha=.6)
    plt.title(title, fontsize=30, fontweight="bold")

    plt.rc('font', size=20)
    plt.axis('off')

    return plt

plt = plot_treemap(df_std_1, 1)
plt.show()

0 个答案:

没有答案