loop over 2d subplot as if it's a 1-D

Question

I\'m trying to plot many data using subplots and I\'m NOT in trouble but I\'m wondering if there is a convenience method to do this.

below is the sample code.

Answer 1

subplots returns an ndarray of axes objects, you can just flatten or ravel it:

fig, axes = plt.subplots(nrows = nrow, ncols=ncol, figsize=(8,6))
for ax in axes.flatten()[:20]:
    # do stuff to ax