python - 如何用 Python 中的其他多项式替换多项式中的 x？

Question

如何用 Python 中的其他多项式替换多项式中的 x？例如，我有 p = poly1d([1,1,1])，这是 x^2+x+1，现在我想替换 x 由 y^2 得到 y^4+y^2+1。这就像两个函数的组合。

Answer 1

您可以使用 numpy.polyval使用 NumPy 组合多项式。例如:

import numpy as np

p1 = np.poly1d([1, 1, 1])
print (p1)
p2 = np.poly1d([1, 0, 0], variable = 'y')
print (p2)
p = np.polyval(p1, p2)
p = np.poly1d(p, variable = 'y')
print (p)

将打印:

   2
1 x + 1 x + 1
   2
1 y
   4     2
1 y + 1 y + 1