有没有一种简单/干净的方式可以迭代子图返回的轴数组,比如:
nrow = ncol = 2
a = []
fig, axs = plt.subplots(nrows=nrow, ncols=ncol)
for i, row in enumerate(axs):
for j, ax in enumerate(row):
a.append(ax)
for i, ax in enumerate(a):
ax.set_ylabel(str(i))
这甚至适用于 nrow
或 ncol == 1
。
我尝试了列表推导式,例如:
[element for tupl in tupleOfTuples for element in tupl]
但如果 nrows
或 ncols == 1
,那么这种方法就失败了。