复数n次方根计算中for循环遍历异常问题求助

我明白你遇到的麻烦了——你的代码因为循环逻辑出错，只能算出一个错误结果，而不是复数n次方根本该有的n个解。让我帮你拆解问题并修正代码：

核心问题分析 #

你原来的代码里，kkk列表是在while循环内部初始化的，每次循环都会清空并只添加当前的counter值，导致for k in kkk每次都只遍历一个元素。而且counter从1到deg的取值也不符合复数方根公式里k的范围（正确应该是0到deg-1）。另外，辐角θ的计算逻辑绕了弯路，很容易出错。

复数n次方根的正确公式 #

对于复数 ( z = r(\cos\theta + i\sin\theta) )，它的n个n次方根遵循这个公式：
[ z_k = r^{1/n} \left( \cos\left( \frac{\theta + 2\pi k}{n} \right) + i\sin\left( \frac{\theta + 2\pi k}{n} \right) \right) ]
其中 ( k = 0,1,...,n-1 )

修正后的Python代码 #

import math

# 获取用户输入
deg = int(input("entrez le degré: "))
re = int(input("le réel: "))
im = int(input("l'imaginaire: "))

# 计算复数的模r和辐角θ（弧度制）
r = math.sqrt(re**2 + im**2)
theta = math.atan2(im, re)  # 直接用atan2获取正确辐角，范围是[-π, π]

# 计算模的n次方根
r_root = r ** (1/deg)

print(f"复数 {re} + {im}i 的 {deg} 次方根如下：")
# 遍历所有k值，生成每个次方根
for k in range(deg):
    # 计算当前根的辐角
    root_theta = (theta + 2 * math.pi * k) / deg
    # 极坐标形式输出
    print(f"z_{k} = {r_root:.2f} * e^(i*{root_theta:.2f})")
    # 可选：转换成直角坐标形式输出，方便验证
    real_part = r_root * math.cos(root_theta)
    imag_part = r_root * math.sin(root_theta)
    print(f"直角坐标形式：{real_part:.2f} + {imag_part:.2f}i\n")

关键修复点 #

1. 循环逻辑修正：移除了错误的while循环，改用range(deg)直接遍历k从0到deg-1，确保生成n个解。
2. 辐角计算优化：用math.atan2(im, re)直接计算复数的辐角，避免了原来复杂且易出错的转换逻辑。
3. 公式正确应用：严格按照复数n次方根的公式计算每个根的模和辐角。
4. 输出格式优化：使用格式化字符串让输出更清晰，同时提供直角坐标形式的可选输出，方便验证结果。

额外提示 #

如果输入的复数是0（re和im都为0），所有次方根都是0，你可以在代码里添加一个判断分支来处理这种特殊情况，避免不必要的计算。

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