python - 将数据 reshape 为 'closest square'

Question

我是 python 的新手。目前使用 matplotlib 我有一个脚本，它返回要制作的可变数量的子图，我将其传递给另一个脚本来进行绘图。我想将这些子图安排得很好，即“最接近正方形的东西”。所以答案是唯一的，假设我给列数赋予更高的权重

示例:假设我要绘制 6 个图，我需要的网格是 2x3。如果我有 9，它就是 3x3。如果我有 12 个，那就是 3x4。如果我有 17 个，它是 4x5，但最后一行只创建了一个。

尝试解决方案:我可以轻松找到最近的足够大的正方形:

    num_plots = 6
    square_size = ceil(sqrt(num_plots))**2

但这会留下空地。有没有办法制作正确的网格大小？

Answer 1

这是我过去所做的

num_plots = 6
nr = int(num_plots**0.5)
nc = num_plots/nr
if nr*nc < num_plots:
    nr+=1
fig,axs = pyplot.subplots(nr,nc,sharex=True,sharey=True)