☰

自动创建图像网格

tag-icon • tag-icon graphics python
自动创建图像网格

我有几个用 生成的 pdf 图像matplotlib。我希望能够将它们排列在任意网格中。为此，我编写了一个 python 脚本，该脚本依次写入并运行 latex。

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--images', nargs='+', help='List of images to be processed')
    parser.add_argument('--output', help='Output file name', default='output_grid.pdf')
    parser.add_argument('--grid', nargs=2, type=int, help='Grid dimensions (rows, columns)')
    args = parser.parse_args()
    return args

def main():
    args = parse_args()
    assert len(args.images) == args.grid[0]*args.grid[1], "Number of images does not match grid dimensions"
    tex_outfile = "_{}_.tex".format(random.randint(0,100000))
    with open(tex_outfile, 'w') as f:
        f.write("\\documentclass[varwidth]{standalone}\n")
        f.write("\\usepackage[pdftex]{graphicx, color}\n")
        f.write("\\begin{document}\n")
        f.write("\\begin{{tabular}}{{{}}}\n".format(" c "*args.grid[1]))
        for k,v in enumerate(args.images):
            f.write("\\includegraphics[width={:.9f}\\linewidth,keepaspectratio]{{{}}}".format(1.0/(args.grid[1] + 1), v))
            if (k+1) % args.grid[1] != 0:
                f.write(" & ")
            else: 
                f.write(" \\\\\n")
        f.write("\\end{tabular}\n")
        f.write("\\end{document}\n")
    os.system("pdflatex {}".format(tex_outfile))
    os.system("mv {}.pdf {}".format(tex_outfile[:-4], args.output))
    os.system("rm {}*".format(tex_outfile[:-4]))

pdflatex上面的 Python 代码生成下面的 Latex，然后在 Python 代码行中运行os.system("pdflatex {}".format(tex_outfile))

\documentclass[varwidth]{standalone}
\usepackage[pdftex]{graphicx, color}
\begin{document}
\begin{tabular}{ c  c }
\includegraphics[width=0.333333333\linewidth,keepaspectratio]{image1.pdf} & \includegraphics[width=0.333333333\linewidth,keepaspectratio]{image2.pdf} \\
\includegraphics[width=0.333333333\linewidth,keepaspectratio]{image3.pdf} & \includegraphics[width=0.333333333\linewidth,keepaspectratio]{image4.pdf} \\
\end{tabular}
\end{document}

我遇到的问题是图像没有完全对齐，请看此图像（已转换为 png 以供 stackexchange 使用）：

在此处输入图片描述

我认为对齐问题是由于 y 轴刻度长度不同造成的。有没有办法在我提供的代码中修复这个问题？或者有一种更好的替代方法（使用subplots将不起作用，因为单个图像已经以 pdf 格式提供）。

答案1

右对齐而不是居中，这样左标签长度不会影响事物，并使用比例而不是宽度，因此图像按相同的量缩放，因此如果它们最初大小相同，则图将具有相同的大小。

相关内容