在循环中使用多个 `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()