python - 如何为子图提供两行多列

Question

以下代码生成下面的图像，其中包含一行多张图像。我想在当前行的下方或上方添加另一行，其中包含数组中的另一系列图像，test类似于predict数组。我为子图尝试了几个数字，比如 2 而不是 1，但它只会给出错误或不起作用。如果我用 2 替换 1 它会给出AttributeError: 'numpy.ndarray' object has no attribute 'matshow'第 4 行的错误，这似乎无关紧要！

你能指导我该怎么做吗？

fig, ax = plt.subplots(1, len(predict), figsize=(10, 5))
fig.suptitle('Output Data with bias term {0}'.format(b))
for i in range(len(predict)):
    ax[i].matshow(predict[i].reshape((n_units, n_units)), cmap='gray')
    ax[i].set_xticks([])
    ax[i].set_yticks([])
plt.show('{0:02d}'.format(b))

标签： pythonsubplot