Python中如何从numpy多项式中提取系数向量？

NumPy提供了两种主要的多项式构造方式，对应的系数提取方法略有不同，下面分别说明：

1. 使用numpy.poly1d构造的多项式 #

当你用numpy.poly1d创建多项式对象后，直接访问其coeffs属性就能得到系数向量，注意这里的系数是从最高次项到常数项排列。

示例代码：

import numpy as np

# 构造多项式 3x² + 2x + 1
poly = np.poly1d([3, 2, 1])
# 提取系数向量
coeff_vector = poly.coeffs
print(coeff_vector)  # 输出: [3 2 1]

2. 使用numpy.polynomial模块的多项式类 #

numpy.polynomial下的各类多项式（如Polynomial标准多项式、Chebyshev切比雪夫多项式等），都通过coef属性提取系数，这里的系数是从常数项到最高次项排列。

标准多项式示例 #

from numpy.polynomial import Polynomial

# 构造多项式 1 + 2x + 3x²
poly = Polynomial([1, 2, 3])
# 提取系数向量
coeff_vector = poly.coef
print(coeff_vector)  # 输出: [1 2 3]

其他多项式基的系数转换 #

如果使用的是非标准多项式基（如切比雪夫），若需要转为标准多项式系数，可以先用convert()方法转换类型，再提取coef：

from numpy.polynomial import Chebyshev, Polynomial

# 构造切比雪夫多项式
cheb_poly = Chebyshev([1, 2, 3])
# 提取切比雪夫基系数
cheb_coeffs = cheb_poly.coef
print(cheb_coeffs)  # 输出: [1 2 3]

# 转换为标准多项式并提取系数
standard_coeffs = cheb_poly.convert(kind=Polynomial).coef
print(standard_coeffs)

内容的提问来源于stack exchange，提问作者Bastiaan Quast