如何在Numpy中自定义多项式打印以剔除近似零系数？

方法1：自定义打印函数（推荐） #

直接遍历多项式的系数与对应次数，过滤掉绝对值小于阈值的项，生成符合需求的打印字符串：

import numpy as np

def print_polynomial(poly, threshold=1e-10):
    valid_terms = []
    for degree, coeff in enumerate(poly.coef):
        if np.abs(coeff) < threshold:
            continue
        # 生成对应次数的项字符串
        if degree == 0:
            term = f"{coeff}"
        elif degree == 1:
            term = f"{coeff}·x¹"
        else:
            term = f"{coeff}·x^{degree}"
        valid_terms.append(term)
    
    if not valid_terms:
        print("p =  0")
    else:
        print(f"p =  {' + '.join(valid_terms)}")

# 测试示例
p = np.polynomial.Polynomial([1.23456789e-15, 1.0, 1.23456789e-13, 2.0])
print_polynomial(p)

输出结果：

p =  1.0·x¹ + 2.0·x³

说明 #

• 可根据数值积分的精度调整threshold阈值（如1e-12）；
• 若需格式化系数显示（如保留固定小数位数），可将f"{coeff}"替换为f"{coeff:.2f}"这类格式；
• 函数会自动过滤所有绝对值小于阈值的项，包括开头、中间的小系数。

方法2：清理系数后构造新多项式（适用于后续计算） #

如果需要在打印之外继续使用清理后的多项式，可以先将小系数置为0，再构造新的多项式：

import numpy as np

p = np.polynomial.Polynomial([1.23456789e-15, 1.0, 1.23456789e-13, 2.0])
threshold = 1e-10

# 将小于阈值的系数替换为0
clean_coeffs = np.where(np.abs(p.coef) < threshold, 0, p.coef)
clean_p = np.polynomial.Polynomial(clean_coeffs)

# 仍需配合自定义打印函数来隐藏零系数项
print_polynomial(clean_p)

注意 #

NumPy的Polynomial.trim()方法仅能去除最高次的零系数，无法处理中间或开头的零项，因此单独使用trim无法满足需求，必须先过滤小系数再处理。

内容的提问来源于stack exchange，提问作者Michael Baudin