如何从函数返回复数？解决Python虚数赋值语法错误

问题1：如何从函数返回复数？ #

在Python里返回复数非常直接，你可以像返回普通数值一样，直接返回一个复数表达式，或者用complex()构造函数生成复数后返回。举两个简单的例子：

方式1：直接返回实部+虚部的表达式 #

def calculate_complex():
    # 计算实部和虚部
    real = 5
    imag = 3
    # 直接返回复数
    return real + imag * 1j

调用这个函数后，你就能得到一个复数：

result = calculate_complex()
print(result)  # 输出 (5+3j)

方式2：使用complex()构造函数 #

complex()接受两个参数，分别是实部和虚部，用它构造复数后返回也很方便：

def get_complex_number(real_part, imag_part):
    return complex(real_part, imag_part)

# 调用示例
my_complex = get_complex_number(2, -4)
print(my_complex)  # 输出 (2-4j)

问题2：解决虚数赋值的语法错误 #

你遇到的SyntaxError是因为Python的语法规则：虚数后缀j只能直接跟在数值字面量后面（比如2j、3.5j），不能直接跟在函数调用或表达式后面——解析器无法识别sqrt(4)j这种写法。

这里有几种可行的解决方法：

方法1：用乘法明确表示虚数部分 #

把函数调用的结果乘以1j，这样解析器就能正确理解你要生成虚数：

import math
a = math.sqrt(4) * 1j
print(a)  # 输出 2j

方法2：先计算结果再赋值 #

先把sqrt(4)的结果存到一个变量里，再和1j结合，可读性更好：

import math
sqrt_result = math.sqrt(4)
a = sqrt_result * 1j
print(a)

方法3：用complex()构造函数 #

如果需要更清晰的语义，可以用complex()，把实部设为0，虚部传入计算结果：

import math
a = complex(0, math.sqrt(4))
print(a)  # 输出 2j

为什么a = 2j是合法的？因为2是一个数值字面量，Python的语法允许字面量直接跟j来表示纯虚数，而函数调用是一个表达式，必须通过乘法或者构造函数来明确生成复数。

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