Python输出格式问题求助：对输出语法存疑，附相关代码

嘿，我看你在处理复系数多项式的输出格式问题，这段代码里的输出逻辑其实不难捋清楚，我来帮你拆解下，顺便给点优化建议：

首先先把你的代码贴出来方便讨论：

import numpy
def g(): #生成随机复数值
    return numpy.random.random(1) + numpy.random.random(1)*1j
p = numpy.poly1d(numpy.squeeze([g(),g(),g()])) #测试函数p
pprime = numpy.polyder(p) #p的导数
print 'Our p(x) is {} '.format(p)
print('\n') #换行
print'Our pprime(x) is {} '.format(pprime)
#对p应用牛顿法
print('\n') #换行
#对...应用牛顿法

首先说现有输出的语法逻辑 #

1. numpy.poly1d的默认输出：当你把p（一个poly1d对象）传给format方法时，numpy会自动调用它的__str__方法，生成类似常规多项式的字符串，比如复数系数的话会显示成(0.45+0.23j)x² + (0.78-0.11j)x + 0.66这样的形式，这个是numpy内置的格式化逻辑，不用额外处理就能看懂。
2. 换行的实现：你用print('\n')来输出空行，这个在Python 2里是没问题的，但如果是Python 3的话，print是函数，得加括号写成print('\n')；另外其实可以更简洁——直接在输出多项式的print语句里指定end='\n\n'，比如print('Our p(x) is {} '.format(p), end='\n\n')，这样一次print就能完成内容输出加换行，不用单独调用一次print。
3. Python版本兼容问题：你的代码里用了print 'xxx'这种Python 2的语法，如果在Python 3环境下运行会报错，建议改成函数形式print('Our p(x) is {} '.format(p))，兼容性更好。

如果想自定义输出格式 #

如果你觉得默认的输出不够美观，或者需要更规整的复数显示（比如固定小数位数），可以自己写个格式化函数，利用poly1d的coeffs属性获取所有系数，然后拼接成想要的样式：

def format_complex_poly(poly):
    terms = []
    max_degree = len(poly.coeffs) - 1  # 多项式最高次项次数
    for idx, coeff in enumerate(poly.coeffs):
        current_degree = max_degree - idx
        # 格式化复数系数，保留两位小数，虚部为正的话显示+号
        if coeff.imag != 0:
            coeff_str = f"{coeff.real:.2f}{coeff.imag:+.2f}j"
        else:
            coeff_str = f"{coeff.real:.2f}"
        # 根据次数生成项的字符串
        if current_degree == 0:
            term = coeff_str
        elif current_degree == 1:
            term = f"{coeff_str}x"
        else:
            term = f"{coeff_str}x^{current_degree}"
        terms.append(term)
    return " + ".join(terms)

# 调用自定义函数输出
print(f"Our p(x) is {format_complex_poly(p)}", end='\n\n')
print(f"Our pprime(x) is {format_complex_poly(pprime)}", end='\n\n')

后续牛顿法的输出建议 #

如果之后要实现牛顿法的迭代过程，输出每一步的迭代值，同样可以用格式化字符串来规范复数的显示，比如：

# 示例牛顿法迭代输出（假设初始值为0）
x_current = 0 + 0j
for step in range(5):
    x_next = x_current - p(x_current)/pprime(x_current)
    print(f"Step {step+1}: x = {x_next.real:.4f}{x_next.imag:+.4f}j")
    x_current = x_next

这样输出的每一步迭代值都会非常规整，方便查看收敛情况。

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