从多个函数调用中叠加 matplotlib 图，同时分别绘制它们

Question

我有一个函数，它使用 seaborn 的 distplot 绘制和显示分布。看起来像这样

def getPlot(data):
    x=sns.distplot(data, hist=False)
    plt.show()
    return x

每次调用该函数时，我都会得到分布图。

我需要一些帮助来修改函数，以便在多次调用函数结束时我应该得到一个额外的图，它是所有先前图的组合。

如果我的函数调用是

getPlot(data1)
getPlot(data2)
getPlot(data3)

我应该在调用函数时获取数据的各个图，并且在最后我希望将 3 个数据的图相互叠加。

仅将 plt.show() 移到函数外是不够的，因为我想要单独数据的单独图以及包含所有数据的图。

Answer 1

由于您必须同时保留一个单独的图 和 一个联合的数据，因此您必须将每个数据集绘制两次。一次在一个单独的轴上，一次在一个共同的轴上。

我要做的是创建一个图形和一个轴，将所有内容绘制在一起。然后将该轴对象传递给函数，并将函数绘制到轴以及新图形中：

def plot_twice(data, ax_all):
    # first plot into the common axes
    sns.distplot(data, hist=False, ax=ax_all)

    # and create a new figure and axes for a standalone plot
    fig,ax = plt.subplots()
    x = sns.distplot(data, hist=False, ax=ax)
    return x

# create axes for the common plot
fig,ax_all = plt.subplots()

# now plot the things
getPlot(data1, ax_all)
getPlot(data2, ax_all)
getPlot(data3, ax_all)

# only call blocking plt.show() at the end
plt.show()

使用 matplotlib 将绘图从一个轴复制 到另一个轴似乎并不可行（参见例如 this 或 this），因此除非绘图花费过多时间或为简单起见，我将数据绘制两次。