按列填充matplotlib子图，而不是行

Question

默认情况下，matplotlib子图由行填充，而不是按列填充。澄清一下，命令

plt.subplot(nrows=3, ncols=2, idx=2)
plt.subplot(nrows=3, ncols=2, idx=3)

首先绘制到3x2绘图网格（idx = 2）的右上图，然后进入左中图（idx = 3）。

有时，无论出于何种原因，可能需要按行填充子图，而不是按列填充（例如，因为直接连续的图集在一起并且当位于彼此下方时更容易解释，而不是彼此相邻）。怎么能实现这一目标？

Answer 1

您可以使用以下命令创建3x2轴阵列：

fig, axes = plt.subplots(nrows=3, ncols=2)

如果您移调此数组，然后展平，您可以按列方式绘制，而不是按行绘制：

fig, axes = plt.subplots(nrows=3, ncols=2)

for ax in axes.T.flatten():
    ax.plot([1,2,3])

Answer 2

可能有一百万（简单）方法可以做到这一点，但由于我花了一分多钟才考虑这个问题，我想我会分享我的解决方案：

def row_based_idx(num_rows, num_cols, idx):
    return np.arange(1, num_rows*num_cols + 1).reshape((num_rows, num_cols)).transpose().flatten()[idx-1]

有了这个，人们就可以做到

row_based_plot_idx = row_based_idx(num_rows, num_cols, col_based_plot_idx)
plt.subplot(num_rows, num_cols, row_based_plot_idx)

希望这能节省一分钟。当然，有更好的解决方案。