python - 如何漂亮地打印到 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)

该代码将生成一个宽度有限的输出（我讨厌）：

我的问题是：

1. 为什么在我的窗口有足够空间的情况下存在宽度限制以及如何更改它？
2. 如何将这样的输出打印到文件中？

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。