Python中输出复数模精确根式及含根式复数输入处理问题

问题说明 #

• 用numpy计算1+1j的模时，输出是近似小数1.4142135623730951，想要√2这种精确根式形式，试了SymPy没成功，求简单实现方式；
• 当前代码能解析1+i这类复数，但输入sqrt(3)-1i或1-sqrt(3)i就报错，得改代码支持带根式的输入，还要保留把i转j适配numpy角度计算的逻辑。

用户提供的代码 #

初始代码 #

import numpy

complexnr = 1+1j
mod= numpy.absolute(complexnr)
print(mod)

编辑后的代码片段 #

input_list = ["Enter your complex number (a+bi): ", \
                      "Degrees or radians?", \
                      "To how many decimal places do you want to round the argument?"]
output = multenterbox(text, title, input_list)

algebraline = output[0]
choice = output[1]
round2 = int(output[2])

#converting complex number to a suitable form for numpy
if "i" in algebraline:
    j = algebraline.replace("i","j")
    indeks = algebraline.index("i")
    list = []
    for element in algebraline:
        list.append(element)
    if "i" in algebraline and algebraline[indeks-1]=="+" or algebraline[indeks-1]=="-":
        list.insert(indeks, 1)
        x = "".join(str(e) for e in list)
        j = x.replace("i","j")
    arv = eval(j)
elif "i" not in algebraline:
    arv = eval(algebraline)

#let's find the module
a = int(list[0])
b = int(list[2])
module = sqrt(a**2+b**2)

解决方法 #

1. 输出精确根式形式的模 #

用SymPy完全可以实现，之前没成功大概率是用法不对。直接用SymPy定义复数、计算模并化简即可：

from sympy import Complex, sqrt, simplify

# 定义复数
z = Complex(1, 1)
# 计算并化简模
exact_mod = simplify(sqrt(z.real**2 + z.imag**2))
print(exact_mod)  # 输出 sqrt(2)

如果要处理用户输入的含根式复数，用sympify解析输入更方便：

from sympy import sympify, sqrt, simplify

user_input = "sqrt(3)-1i"
# SymPy用I表示虚数单位，先替换输入里的i
z = sympify(user_input.replace('i', 'I'))
# 计算精确模
exact_mod = simplify(sqrt(z.real**2 + z.imag**2))
print(exact_mod)  # 输入sqrt(3)-1i时，输出2

2. 修复含根式的输入解析问题 #

原代码的问题在于：只处理了i前是+/-的简单情况，没考虑根式后跟i的情况；而且用int(list[0])提取实部虚部完全错误，因为输入可能是根式或非整数。

修改后的解析代码：

import re
from sympy import sympify
import numpy as np

algebraline = "1-sqrt(3)i"  # 示例输入

# 先补全i前的缺失项：比如把+i变成+1i，sqrt(3)i变成sqrt(3)*i
algebraline = re.sub(r'(^|[+-])i', r'\11i', algebraline)  # 补全±i前的1
algebraline = re.sub(r'([0-9)])i', r'\1*i', algebraline)  # 补全数字/括号后的乘号

# 用SymPy解析为符号复数
z_sym = sympify(algebraline.replace('i', 'I'))
# 转成numpy兼容的复数（如果需要数值计算角度）
arv = complex(z_sym.evalf())  # 转为Python内置复数，numpy可直接使用

# 计算精确模
exact_mod = simplify(sqrt(z_sym.real**2 + z_sym.imag**2))
print("精确模：", exact_mod)
# 用numpy计算角度（弧度）
angle_rad = np.angle(arv)
print("弧度角度：", angle_rad)

核心改进：

• 用正则表达式自动补全i前缺失的系数和乘号，解决格式不规范的问题；
• 用SymPy的sympify解析含根式的表达式，既安全又支持符号运算；
• 同时兼顾精确根式输出和numpy的数值计算需求，不用再生硬拆分字符串提取实部虚部。

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