for循环中有多个`subplot2grid`

Question

我试图通过双循环绘制两个不同的函数集。我不知道如何让subplot2grid对第二个数字采取行动。

import numpy as np
from matplotlib import pyplot as plt
t=np.linspace(0,1,100)
fig1=plt.figure()
for i in range(3):
    for j in range(3):
        plt.subplot2grid((3,3),(i,j))
        plt.plot(t,np.sin((t*np.random.random()*10)))
fig2=plt.figure()
for i in range(3):
    for j in range(3):
        plt.subplot2grid((3,3),(i,j))
        plt.plot(t,np.cos((t*np.random.random()*10)))
plt.show()

有没有办法只用一个循环来做到这一点？

Answer 1

您应该使用面向对象的界面。这是一个例子：

import numpy as np
from matplotlib import pyplot as plt

t = np.linspace(0, 1, 100)
fig1, axes1 = plt.subplots(3, 3)
fig2, axes2 = plt.subplots(3, 3)

blue, red = "#1E90FF", "#FF6347"
for i in range(3):
    for j in range(3):
        axes1[i, j].plot(t, np.sin((t * np.random.random() * 10)), blue)
        axes2[i, j].plot(t, np.cos((t * np.random.random() * 10)), red)

fig1.tight_layout()
fig2.tight_layout()