如何创建返回pyplot对象且仅显式调用时绘图的函数

要实现「创建函数返回matplotlib图形对象、默认不自动绘图、显式触发才显示」的需求，你可以通过以下方式修改代码：

修改后的函数 #

import numpy as np
import matplotlib.pyplot as plt

def plot_it_now(show=False):
    # 生成数据
    A = np.linspace(0, 10, 10)
    B = np.exp(A)
    
    # 创建Figure对象，用面向对象方式绘图（更可控）
    fig = plt.figure()
    ax = fig.add_subplot(111)
    
    # 绘制图形并设置元素
    ax.plot(A, label='linear sequence')
    ax.plot(B, label='exponential sequence')
    ax.set_xlabel('x-axis')
    ax.set_ylabel('y-axis')
    ax.legend(loc='best')
    
    # 仅当显式指定show=True时才立即显示
    if show:
        plt.show()
    
    # 返回Figure对象，供后续操作
    return fig

关键改动说明 #

• 返回Figure对象：原函数没有返回值，你之前赋值的myplot其实是None。现在函数返回创建的fig对象，你可以将其保存到变量中，后续随时修改或显示。
• 面向对象式绘图：用ax.plot替代全局的plt.plot，避免依赖matplotlib的全局状态，让图形控制更精准。
• 可控显示开关：添加show可选参数，默认False，只有显式传入show=True才会立即显示图形；否则图形只会在你主动调用显示方法时才渲染。

使用示例 #

1. 仅保存图形对象，后续按需显示：

# 保存图形到变量，此时不会自动显示
myplot = plot_it_now()

# 可以对图形做后续修改，比如调整尺寸
myplot.set_figsize((8, 6))

# 需要显示时手动调用
plt.show()

2. 显式指定立即显示：

# 调用函数时直接传入show=True，立即显示图形
myplot = plot_it_now(show=True)

3. 直接通过Figure对象显示：

myplot = plot_it_now()
# 调用Figure对象自身的show方法
myplot.show()

关于自动显示的说明 #

你之前遇到的「赋值时自动绘图」，是因为matplotlib的交互式后端（比如Jupyter的inline模式、Qt后端）会在创建Figure时自动渲染。通过上述方式，我们将显示的控制权完全交给你，避免自动触发渲染。

内容的提问来源于stack exchange，提问作者Aman Singh