Как красиво печатать в файл в симпы? - PullRequest
0 голосов
/ 30 мая 2018

Предположим, у меня есть следующий код:

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)

Код сгенерирует вывод с ограниченной шириной (который я ненавижу):

enter image description here

Мои вопросы:

  1. Почему существует ограничение ширины, когда в моем окне достаточно места и как его изменить?
  2. Как напечатать такой вывод вфайл?

1 Ответ

0 голосов
/ 30 мая 2018

Для 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)

.

...