将 matplotlib 图例保存为单独的图像

Question

在发布问题之前我浏览了以下链接。

在 Matplotlib 中将图例作为单独的图片获取

我无法为我的示例复制提供的解决方案。

这是我的数据框和代码。我想将图例保存为单独的图像。

df=pd.DataFrame(index=['A','B','C','D'], columns=['Values'])
df['Values'] = [0.45,0.28,0.21,0.3]

fig=plt.figure(figsize=(8,8))
ax1 = plt.subplot(121, aspect='equal')
df['Values'].dropna().plot(kind='pie', autopct='%1.0f%%', startangle=220, labels=None, 
                           colors=['#002c4b','#392e2c','#92847a','#ccc2bb','#6b879d'])

patches, labels = ax1.get_legend_handles_labels()
pp = ax1.legend(patches, labels=df.index, loc='center right', bbox_to_anchor=(0, 0.5), 
              fontsize=8, frameon=False, labelspacing=4)
ax1.axis('off')
plt.ylabel('')
img_name = 'unit1.png'   
plt.savefig(img_name,bbox_inches='tight', dpi = 300)
plt.close(fig)

请提出前进的方向。

Answer 1

不知何故，pandas 到 matplotlib 的接口似乎搞乱了对get_legend_handles_labels. 下面的代码与您编写的代码等效，但使用 matplotlib 的接口。

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame(index=['A','B','C','D'], columns=['Values'])
df['Values'] = [0.45, 0.28, 0.21, 0.3]

fig, ax = plt.subplots()
ax.pie(df["Values"].dropna(), 
       colors=['#002c4b', '#392e2c', '#92847a', '#ccc2bb', '#6b879d'], 
       autopct='%1.0f%%',
       startangle=220,
       labels=df.index,)
ax.legend(loc="best")

# get handles and labels for reuse
label_params = ax.get_legend_handles_labels() 

figl, axl = plt.subplots()
axl.axis(False)
axl.legend(*label_params, loc="center", bbox_to_anchor=(0.5, 0.5), prop={"size":50})
figl.savefig("LABEL_ONLY.png")

您可能需要使用figaspect和prop来获得标签尺寸和位置的最佳结果。