python - 当行数或列数为单位时，使用两个索引来引用子图轴

Question

可以使用两个索引来索引不同的子图的轴，如下例所示

rows = 2
cols = 2
f, ax = plt.subplots(rows, cols)
x = np.arange(12)
y = xdata**2
plotFunction(x,y,ax,0,1)

def plotFunction(xdata, ydata, ax, i, j):
    ax[i,j].plot(xdata, ydata, marker='o', label='quadratic')

但是，如果 rows 或 cols = 1 pyplot 不允许使用两个索引。这阻止了依赖双索引绘图的绘图函数的通用使用。所以下面的方法是行不通的

rows = 1
cols = 2
f, ax = plt.subplots(rows, cols)
x = np.arange(12)
y = xdata**2
plotFunction(x,y,ax,0,1)

Answer 1

一种方法是在调用子图时使用“squeeze=False”选项。

rows = 1
cols = 2
f, ax = plt.subplots(rows, cols, squeeze=False)
x = np.arange(12)
y = xdata**2
plotFunction(x,y,ax,0,1,label='quadratic')

def plotFunction(xdata, ydata, ax, i, j, label):
ax[i,j].plot(xdata, ydata, marker='o', label=label)

这允许在所有情况下建立 [row,col] 索引。