无限期绘制matplotlib子图？

Question

say I was testing a range of parameters of a clustering algorithm and I wanted to write python code that would plot all the results of the algorithm in subplots 2 to a row

is there a way to do this without pre-calculating how many total plots you would need?

something like:

for c in range(3,10):
    k = KMeans(n_clusters=c)
    plt.subplots(_, 2, _)
    plt.scatter(data=data, x='x', y='y', c=k.fit_predict(data))

... and then it would just plot 'data' with 'c' clusters 2 plots per row until it ran out of stuff to plot.

thanks!

Answer 1

This answer from the question Dynamically add/create subplots in matplotlib explains a way to do it: https://stackoverflow.com/a/29962074/3827277

verbatim copy & paste:

import matplotlib.pyplot as plt

# Start with one
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot([1,2,3])

# Now later you get a new subplot; change the geometry of the existing
n = len(fig.axes)
for i in range(n):
    fig.axes[i].change_geometry(n+1, 1, i+1)

# Add the new
ax = fig.add_subplot(n+1, 1, n+1)
ax.plot([4,5,6])

plt.show()