为多个图表重新使用轴对象

Question

我有一个带有值的pandas DataFrame和一个季节指示器。 我想为整个场景创建一个密度图，然后为每个季节创建一个密度图，其中应包括总体密度加上季节的密度。
由于总体密度估算需要一些时间（我有超过40000个数据点），因此我想避免每个季节重复进行估算。

到目前为止，基于此答案 ，我得到的最接近的代码是：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

n = 25
seasons = ['winter', 'spring', 'summer', 'autumn']
df = pd.DataFrame({'value': np.random.randn(n), 'season': np.random.choice(seasons, n)})

ax = df['value'].plot.density(label="all", legend=True)
plt.savefig("test_density_all.png")
for s in seasons:
    sdf = df[df['season'] == s]
    sdf['value'].plot.density(label=s, legend=True)
    plt.savefig("test_density_" + s + ".png")
    del ax.lines[-1]

在这里，我要保存的是删除季节线，而不是保存整个情节。 问题在于它不会删除图例，因此夏季图具有正确的两个密度时，其图例包括四个项目（全部，冬季，春季，夏季）。

因此，我需要的是使以上代码与图例一起使用，或者找到一种方式来存储整体图，以便我可以将其用作每个季节图的起点...

Answer 1

使用ax.legend()获取图中的图例。
请注意，这与使用legend=True参数不同，因为它会根据情节中当前存在的艺术家创建新的图例。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

n = 25
seasons = ['winter', 'spring', 'summer', 'autumn']
df = pd.DataFrame({'value': np.random.randn(n), 'season': np.random.choice(seasons, n)})

ax = df['value'].plot.density(label="all", legend=True)
plt.savefig("test_density_all.png")
for s in seasons:
    sdf = df[df['season'] == s]
    sdf['value'].plot.density(label=s)
    ax.legend()
    ax.figure.savefig("test_density_" + s + ".png")
    ax.lines[-1].remove()