如何解决Python中运算结果出现+0j的问题

问题根源 #

你代码里用了cmath.sqrt，它始终返回复数类型，哪怕计算结果的虚部为0，也会显示+0j。另外你的条件判断逻辑存在错误，比如if function == "*" or "multiplication"这种写法等价于if (function == "*") or True，会导致不管输入什么function都会进入该分支，属于潜在bug。

修复方案 #

1. 替换复数平方根为实数版本 #

把from cmath import sqrt改成from math import sqrt，math.sqrt仅处理非负实数，返回浮点数（若为完全平方数可转成整数），不会产生复数类型结果。

2. 修正条件判断逻辑 #

将错误的或运算改成统一小写后的成员判断，简化大小写兼容处理：

• 原if function == "*" or "multiplication" or "MULTIPLICATION" or "Multiplication"改为if function.lower() in ["*", "multiplication"]
• 原if exp == "yes" or "Yes" or "YES"改为if exp.lower() == "yes"

3. 补全未定义变量的分支 #

如果用户输入root不是"yes"，原代码中c和d会未定义导致报错，需补充else分支处理该情况。

修复后的完整代码 #

from math import sqrt  # 替换为math库的实数平方根函数
a = int(input("insert number: "))
b = int(input("insert number: "))
exp = input("Would you like to have exponents? ")
function = input("Put what kind of equation you want: ")

# 修正条件判断，统一转小写简化大小写兼容
if function.lower() in ["*", "multiplication"]:
    if exp.lower() == "yes":
        e = int(input("What number do you want?"))
        root = input("Would you like square root? ")
        if root.lower() == "yes":
            product = a * b
            c = sqrt(product)
            # 若为完全平方数，转成整数更美观，否则保留浮点数
            c = int(c) if c.is_integer() else c
            d = c ** e
            print(a, "was", function, "by", b, "getting the answer of", c)
        else:
            # 不做平方根时，直接计算乘积的幂次
            c = a * b
            d = c ** e
            print(a, "was", function, "by", b, "getting the answer of", c)
        print("Then", c, "was put to the power of", e, "giving the final answer of", d)

额外说明 #

• math.sqrt要求输入为非负实数，由于你输入的是整数，相乘结果必然非负，无需额外处理异常。
• 加入c.is_integer()判断是为了让结果更直观，比如sqrt(4)得到2.0，转成整数后显示2而非2.0。

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