如何在python中使用for循环绘制两个以上的绘图？

import matplotlib.pyplot as plt

# 创建一个包含多个图形的列表
fig, axs = plt.subplots(2, 2)

# 使用for循环遍历图形列表，并绘制图形
for i, ax in enumerate(axs.flatten()):
    # 绘制第一个图形
    if i == 0:
        ax.plot([1, 2, 3, 4], [1, 4, 9, 16], 'r')
        ax.set_title('Plot 1')
    # 绘制第二个图形
    elif i == 1:
        ax.plot([1, 2, 3, 4], [1, 8, 27, 64], 'g')
        ax.set_title('Plot 2')
    # 绘制第三个图形
    elif i == 2:
        ax.plot([1, 2, 3, 4], [1, 10, 100, 1000], 'b')
        ax.set_title('Plot 3')
    # 绘制第四个图形
    elif i == 3:
        ax.plot([1, 2, 3, 4], [1, 2, 3, 4], 'y')
        ax.set_title('Plot 4')

# 显示图形
plt.show()