如何在SymPy中编写方程组并输出对应的LaTeX代码？

SymPy的latex()函数支持直接处理表达式列表，并非只能传入单个表达式，以下是几种实用的实现方式：

方法一：默认生成大括号包裹的方程组 #

直接将方程组以列表形式传入latex()，默认会生成LaTeX的cases环境代码，适合带大括号的方程组展示：

import sympy as sp

x, y, z = sp.symbols('x, y, z')
eq1 = sp.Eq(x + y + z, 1)
eq2 = sp.Eq(x + y + 2 * z, 3)

# 传入方程组列表
output = sp.latex([eq1, eq2])
print(output)

输出的LaTeX代码：

\begin{cases} x + y + z = 1 \\ x + y + 2 z = 3 \end{cases}

方法二：生成等号对齐的方程组 #

如果需要等号对齐的格式，可以指定env='align'参数，生成LaTeX的align环境代码：

output = sp.latex([eq1, eq2], env='align')
print(output)

输出的LaTeX代码：

\begin{align} x + y + z &= 1 \\ x + y + 2 z &= 3 \end{align}

方法三：自定义格式拼接 #

如果需要更灵活的排版（比如自定义分隔符、添加编号等），可以手动遍历每个方程生成LaTeX代码后拼接：

# 逐个生成单个方程的LaTeX代码
latex_eq_list = [sp.latex(eq) for eq in [eq1, eq2]]
# 拼接成自定义的LaTeX环境（示例用align环境）
output = r'\begin{align}' + ' \\\\ '.join(latex_eq_list) + r'\end{align}'

内容的提问来源于stack exchange，提问作者Thegerdfather