matplotlib个子图中的垂直排列

Question

我正在尝试 plot 一个带有 4 个子图的图形。 下面的代码和我写的差不多。

import numpy as np
import matplotlib.pyplot as plt
def trial(x, r, ax=None):
    y = r*np.sin(x)
    ax[0].plot(x, r*x)
    ax[0].set_title('x')
    ax[1].plot(x, y)
    ax[1].set_title('sinx')

x = np.linspace(0, 4*np.pi)    
fig, (ax1, ax2) = plt.subplots(2,2)
trial(x, 1, ax1)
trial(x, 2, ax2)

这是我得到的 output。

如果我只有 plot ax1（从子图中删除最后一个索引后），x 和 sin(x) 是垂直排列的。 但是一旦我添加 ax2，它们就会水平排列。 我希望 x 和 sin(x) 与 ax1 和 ax2 垂直排列。 我应该做什么更正？

Answer 1

我们试试吧：

fig, axes = plt.subplots(2,2)
trial(x, 1, axes[:,0])
trial(x, 2, axes[:,1])

Output：