使用matplotlib进行多重绘图，在运行前不知道绘图的数量

Question

我有问题，matplotlib子图让我感到困惑。我不知道我想事先绘制的子图的数量，但我知道我想要它们分为两行。

所以我无法使用

plt.subplot(212)

因为我不知道我应该提供的号码。

它应该是这样的：

现在我将所有绘图绘制到一个文件夹中并将它们与插图画家放在一起，但是matplotlib必须有更好的方法。如果我某处不清楚，我可以提供我的代码。

Answer 1

我的理解是你只知道运行时的绘图数量，因此正在努力使用速记语法，例如：

plt.subplot(121)

值得庆幸的是，为了节省你不得不做一些笨拙的数学以编程方式计算出这个数字，还有另一个允许你使用表格的界面：

plt.subplot(n_cols, n_rows, plot_num)

所以在你的情况下，如果你想要n图，你可以这样做：

n_plots = 5 # (or however many you programatically figure out you need)
n_cols = 2
n_rows = (n_plots + 1) // n_cols
for plot_num in range(n_plots):
   ax = plt.subplot(n_cols, n_rows, plot_num)
   # ... do some plotting

或者，您可能希望了解更多的pythonic界面：

fig, subplots = plt.subplots(n_cols, n_rows)
for ax in subplots:
   # ... do some plotting

（请注意，这是subplots()而不是普通subplot()）。虽然我必须承认，我从未使用过后一种界面。

HTH