iPython中如何实现复数幅值/角度格式显示及切换

完全可以实现，不需要修改iPython内核，通过重写内置复数类型的__repr__方法就能完成显示逻辑替换，配合切换函数可以在两种格式间灵活跳转，适配草稿式快速计算的使用场景。

完整实现代码 #

把以下代码直接粘贴到当前iPython会话中运行即可，如果需要长期使用，可以放到iPython启动脚本里自动加载：

import cmath
import builtins

# 保存原生复数的默认显示方法，用于切回原格式
_original_complex_repr = builtins.complex.__repr__
_current_display_mode = "rect"  # 初始状态为默认直角坐标格式（实部+虚部j）

def _polar_format_complex(z):
    """极坐标格式化逻辑：输出 幅值 ∠ 角度（角度制）"""
    magnitude = abs(z)
    phase_rad = cmath.phase(z)
    phase_deg = phase_rad * 180 / cmath.pi
    # 处理浮点计算带来的极小值误差，避免输出类似5e-16的无意义数值
    if abs(magnitude) < 1e-10:
        return "0.000 ∠ 0.00°"
    return f"{magnitude:.3f} ∠ {phase_deg:.2f}°"

def toggle_complex_display(mode=None):
    """
    切换复数打印格式
    参数：
        mode: 可选值为'rect'/'polar'，指定切换到对应格式；不传参则在两种格式间循环切换
    """
    global _current_display_mode
    # 处理无参调用的切换逻辑
    if mode is None:
        _current_display_mode = "polar" if _current_display_mode == "rect" else "rect"
    else:
        if mode not in ("rect", "polar"):
            raise ValueError("仅支持两种模式：'rect'为默认实部虚部格式，'polar'为幅值角度格式")
        _current_display_mode = mode
    
    # 替换对应显示方法
    if _current_display_mode == "rect":
        builtins.complex.__repr__ = _original_complex_repr
        print("已切换至默认直角坐标格式：a + bj")
    else:
        builtins.complex.__repr__ = _polar_format_complex
        print("已切换至极坐标格式：幅值 ∠ 角度°")

使用方式 #

• 无参调用toggle_complex_display()：自动在两种显示模式间来回切换
• 指定模式调用：toggle_complex_display("polar")直接切到幅值角度格式，toggle_complex_display("rect")切回默认实部虚部格式

实际运行效果参考：

In [2]: test_z = 1 + 1j

In [3]: toggle_complex_display("polar")
已切换至极坐标格式：幅值 ∠ 角度°

In [4]: test_z
Out[4]: 1.414 ∠ 45.00°

In [5]: toggle_complex_display()
已切换至默认直角坐标格式：a + bj

In [6]: test_z
Out[6]: (1+1j)

自定义调整 #

• 如果需要用弧度显示角度，删掉_polar_format_complex里弧度转角度的换算逻辑，直接用cmath.phase(z)的返回值，把单位后缀改成rad即可
• 显示精度可以自行修改f-string里的小数位数参数，当前默认幅值保留3位小数、角度保留2位小数
• 要实现启动自动加载，把上述代码保存为.py文件，放到iPython对应配置目录的startup文件夹下即可，路径一般为~/.ipython/profile_default/startup/

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