事实证明你可以复制 R 的结果poly(x,p)
通过对矩阵执行 QR 分解来实现该函数,该矩阵的列是输入向量的幂x
从 0 次方(全部)到p
次幂。 Q 矩阵减去第一个常数列,即可得到所需的结果。
因此,以下内容应该有效:
import numpy as np
def poly(x, p):
x = np.array(x)
X = np.transpose(np.vstack((x**k for k in range(p+1))))
return np.linalg.qr(X)[0][:,1:]
尤其:
In [29]: poly([1,2,3,4,5,6,7,8,9,10], 3)
Out[29]:
array([[-0.49543369, 0.52223297, 0.45342519],
[-0.38533732, 0.17407766, -0.15114173],
[-0.27524094, -0.08703883, -0.37785433],
[-0.16514456, -0.26111648, -0.33467098],
[-0.05504819, -0.34815531, -0.12955006],
[ 0.05504819, -0.34815531, 0.12955006],
[ 0.16514456, -0.26111648, 0.33467098],
[ 0.27524094, -0.08703883, 0.37785433],
[ 0.38533732, 0.17407766, 0.15114173],
[ 0.49543369, 0.52223297, -0.45342519]])
In [30]: