如何在函数内while循环结束后基于返回的numpy数组绘制多条matplotlib曲线

核心问题修正 #

先解决代码中的基础语法和逻辑错误：

• 函数调用错误：def是定义函数的关键字，不能作为函数名调用，自定义函数的正确调用名为a
• while循环语法错误：原while:缺少终止条件，会触发无限循环，需补充实际的循环退出判断
• 绘图变量冲突：原代码中B, = plt.plot(t, B)会用matplotlib线对象覆盖原始的numpy数组B，导致数据逻辑错误

重要提示：不要用和原始输入/输出数组同名的变量接收plt.plot的返回值，否则会用matplotlib线对象覆盖原始numpy数组，引发后续逻辑错误。

步骤1：修正函数结构 #

如果需要在函数内部完成绘图，直接在while循环结束后添加绘图逻辑，完整函数示例：

import matplotlib.pyplot as plt

def a(t, A, B, C, At, Bt):
    # 补充while循环的终止条件，替换为你的实际判断逻辑
    while 你的循环终止条件:
        # 你的实际计算逻辑
        calculations
    
    # while循环结束后执行绘图
    # 线对象单独命名，避免覆盖原始numpy数组
    line_B, = plt.plot(t, B)
    line_C, = plt.plot(t, C)
    line_A, = plt.plot(t, A)
    
    # 配置图例，和你需求的格式一致
    plt.legend(handles=[line_B, line_C, line_A],
               labels=['B', 'C', 'A'])
    
    # 可选：显示绘制的图像
    plt.show()
    
    return t, A, B, C, At, Bt

步骤2：正确调用函数 #

替换原来错误的调用方式，根据需求选择调用形式：

• 仅运行函数（包括内部绘图和计算）：a(t, A, B, C, At, Bt)
• 运行函数并打印返回的numpy数组：print(a(t, A, B, C, At, Bt))

补充：如果希望绘图在函数外部完成 #

如果不把绘图逻辑放在函数内部，调用函数后先接收返回值，再执行绘图：

# 调用函数，按return的顺序接收所有返回数组
t_res, A_res, B_res, C_res, At_res, Bt_res = a(t, A, B, C, At, Bt)

# 执行绘图
line_B, = plt.plot(t_res, B_res)
line_C, = plt.plot(t_res, C_res)
line_A, = plt.plot(t_res, A_res)
plt.legend(handles=[line_B, line_C, line_A],
           labels=['B', 'C', 'A'])
plt.show()

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