从函数返回子图

问题描述

在IPython笔记本中使用Matplotlib，我想创建一个带有从函数返回的子图的图形：

import matplotlib.pyplot as plt

%matplotlib inline

def create_subplot(data):
    more_data = do_something_on_data()  
    bp = plt.boxplot(more_data)
    # return boxplot?
    return bp

# make figure with subplots
f, (ax1, ax2) = plt.subplots(1, 2, sharey=True, figsize=(10,5))

ax1 -> how can I get the plot from create_subplot() and put it on ax1?
ax1 -> how can I get the plot from create_subplot() and put it on ax2?

我知道我可以直接将绘图添加到轴：

ax1.boxplot(data)

但是如何从函数返回绘图并将其用作子绘图？

---

解决方案

通常，您会执行如下操作：

def create_subplot(data, ax=None):
    if ax is None:
        ax = plt.gca()
    more_data = do_something_on_data()  
    bp = ax.boxplot(more_data)
    return bp

# make figure with subplots
f, (ax1, ax2) = plt.subplots(1, 2, sharey=True, figsize=(10,5))
create_subplot(data, ax1)

您不能"从函数返回绘图并将其用作子绘图"。相反，您需要将箱图绘制在子图中的轴上。

if ax is None部分就在那里，因此传入显式轴是可选的(如果不是，将使用当前的pyploy轴，与调用plt.boxplot相同)。如果您愿意，可以省略它，并要求指定特定的轴。