python -使用一个函数来绘制多个图？

import matplotlib.pyplot as plt

def plot_multiple_graphs():
    # 准备数据
    x = [1, 2, 3, 4, 5]
    y1 = [1, 4, 9, 16, 25]
    y2 = [1, 8, 27, 64, 125]
    
    # 绘制第一个图表
    plt.subplot(2, 1, 1)  # 创建一个2行1列的子图，当前绘制第1个子图
    plt.plot(x, y1, 'r')  # 绘制红色的折线图
    plt.title('Graph 1')  # 设置图表标题
    
    # 绘制第二个图表
    plt.subplot(2, 1, 2)  # 创建一个2行1列的子图，当前绘制第2个子图
    plt.plot(x, y2, 'b')  # 绘制蓝色的折线图
    plt.title('Graph 2')  # 设置图表标题
    
    # 显示图表
    plt.show()

plot_multiple_graphs()