"如何漂亮地打印到 sympy 中的文件？"

Question

假设我有以下代码：

import sympy as sp
from sympy.physics.quantum import TensorProduct

s=sp.eye(2)
a=TensorProduct(s*x,TensorProduct(s,s)).subs(x,x**2+2*x+1)
sp.pprint(a)

Answer 1

对于python>=3.4 ，

from contextlib import redirect_stdout
import sympy as sp
from sympy.physics.quantum import TensorProduct

s = sp.eye(2)
x = sp.symbols('x')
a = TensorProduct(s*x, TensorProduct(s, s)).subs(x, x**2+2*x+1)

with open('data.txt', 'w') as f:
    with redirect_stdout(f):
        sp.pprint(a, wrap_line=False)

。

Answer 2

sympy.preview ：

sp.preview(a, viewer='file', filename="out.png", dvioptions=['-D','300'])

将允许您将表达式保存为 LaTeX/png/pdf。