python - numpy.polyval() 的反函数

Question

我想知道 np.polyval() 是否有一个方便的反函数，我在其中给出 y 值并求解 x？

我知道我可以这样做的一种方法是:

import numpy as np

# Set up the question
p = np.array([1, 1, -10])
y = 100

# Solve
p_temp = p
p_temp[-1] -= y
x = np.roots(p_temp)

但是我的猜测是大多数人会同意这段代码的可读性低。有什么建议吗？

Answer 1

这样的事情怎么样？

In [19]: p = np.poly1d([1, 1, -10])  # Use a poly1d to represent the polynomial.

In [20]: y = 100

In [21]: (p - y).roots
Out[21]: array([-11.,  10.])

poly1d对象实现算术运算返回一个新的poly1d对象，所以p - y是一个新的poly1d:

In [22]: p - y
Out[22]: poly1d([   1,    1, -110])

poly1d 的 roots 属性返回您所期望的。