python - IPython notebook 中 pretty-print 多项式

Question

我在 IPython notebook 中有一些多项式 x:

import numpy as np
x = np.polynomial.polynomial.Polynomial([1,2,3])
x

然后 x 打印为 Polynomial([ 1., 2., 3.], [-1., 1.], [-1., 1.]) 说实话很丑。我怎样才能正常打印，就像我在 LaTeX 中渲染多项式或将其写在纸上一样？

Answer 1

您可以使用 sympy 的 Poly 类将多项式呈现为漂亮的 latex 。这里唯一的问题是 numpy 按增加程度的顺序列出系数，而 sympy 则相反。

In [1]: import numpy as np
   ...: nppoly = np.polynomial.polynomial.Polynomial([1,2,3])
   ...: nppoly
Out[1]: Polynomial([ 1.,  2.,  3.], [-1.,  1.], [-1.,  1.])

In [2]: import sympy as sp
   ...: from sympy.abc import x
   ...: sp.init_printing()
   ...: sp.Poly(reversed(nppoly.coef),x).as_expr()

给出: