根据列表的长度自动生成子图

Question

我想根据我称之为dets的numpy阵列长度生成色情子图。我想使用数组长度来确定子图的正确数量的列和行。如果是正方形，请绘制子图的正方形矩阵，如果不是正方形，请添加另一个行。对于初学者，我编写了一些代码来检查数组的长度是否会创建子图的正方形矩阵，其中以下是：

data_f = np.random.rand(len(dets),2,5)
dets = np.arange(-5,-0.75,0.25)
x = np.array([1,5,6,3,8,9,2,3,10,12,3])
v = np.linspace(0,10,len(x))

square = np.sqrt(len(dets))
check_square = len(dets)%square
non_square = 1

print(len(data_f))

if check_square == 0:
    nrows = int(np.sqrt(len(dets)))
    ncols = int(np.sqrt(len(dets)))
else:
    nrows = int(np.sqrt(len(dets)))+non_square
    ncols = int(np.sqrt(len(dets)))

fig, ax = plt.subplots(nrows, ncols, sharex='col', sharey='row')

for i in range(nrows):
    for j in range(ncols):
        if i==0:
            im = ax[i,j].imshow(data_f[j],extent=(x.min(), x.max(), v.min(), v.max()),origin='lower',aspect='auto')
        else:
             im = ax[i,j].imshow(data_f[j+ncols*i],extent=(x.min(), x.max(), v.min(), v.max()),origin='lower',aspect='auto')
            

输出图：

该图17个图，但由于错误

这绘制了我想要的所有内容，除了它总是以奇怪的方式将剧情击中，因为有以下错误：

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
~\AppData\Local\Temp\1/ipykernel_4560/3817292743.py in <module>
      6             im = ax[i,j].imshow(data_f[j],extent=(x.min(), x.max(), v.min(), v.max()),origin='lower',aspect='auto')
      7         else:
----> 8              im = ax[i,j].imshow(data_f[j+ncols*i],extent=(x.min(), x.max(), v.min(), v.max()),origin='lower',aspect='auto')
      9 

IndexError: index 17 is out of bounds for axis 0 with size 17

Answer 1

这是因为您总共有20个轴，并且当您循环穿越NCOL和NROWS时，您将获得20次迭代。 data_f）只有17

。

if(j + ncols*i) == len(data_f): 
    break

但是len （