Python内置函数实战(十)：complex函数详解与实战

complex() 是 Python 的内置函数，用于创建或转换复数。复数在科学计算、工程、信号处理等领域广泛应用。以下是关于 complex() 的详细解析和实战案例。

一、基本语法

complex([real[, imag]])

• 参数：
  • real（可选）：实数部分，可为整数、浮点数、字符串（如 "3+4j"），默认为 0。
  • imag（可选）：虚数部分，可为整数或浮点数，默认为 0。

二、核心功能与示例

1. 创建复数的常见方式

# 方式1：无参数（默认值为 0j）
print(complex())           # 输出: 0j

# 方式2：单个数值参数（实数部分）
print(complex(3))          # 输出: 3j
print(complex(2.5))        # 输出: (2.5+0j)

# 方式3：两个数值参数（实数和虚数部分）
print(complex(3, 4))       # 输出: (3+4j)
print(complex(-1, 0.5))    # 输出: (-1+0.5j)

# 方式4：字符串参数（无需指定虚数部分）
print(complex("3+4j"))     # 输出: (3+4j)
print(complex("1.2-2.3j")) # 输出: (1.2-2.3j)

2. 复数的属性和操作

z = complex(3, 4)

# 获取实部和虚部
print(z.real)             # 输出: 3.0
print(z.imag)             # 输出: 4.0

# 计算共轭复数
print(z.conjugate())      # 输出: (3-4j)

# 复数的模（绝对值）
print(abs(z))             # 输出: 5.0（√(3²+4²)）

3. 复数运算

a = complex(2, 3)
b = complex(1, -1)

# 加法
print(a + b)              # 输出: (3+2j)

# 乘法
print(a * b)              # 输出: (5+1j)

# 除法
print(a / b)              # 输出: (-0.5+2.5j)

三、高级用法

1. 从字符串创建复数（注意事项）

# 正确：字符串中必须包含虚数单位 j/J
print(complex("3+4j"))    # 输出: (3+4j)

# 错误：不能使用两个参数的字符串形式
print(complex("3", "4"))  # 报错：TypeError

# 正确：若字符串包含逗号，需先分割处理
s = "3,4"
real, imag = map(float, s.split(','))
print(complex(real, imag))  # 输出: (3+4j)

2. 复数在数学函数中的应用

import cmath  # 复数专用数学模块

z = complex(0, 1)

# 计算指数函数 e^(iπ) = -1
print(cmath.exp(z * cmath.pi))  # 输出: (-1+1.2246467991473532e-16j)

# 计算平方根
print(cmath.sqrt(-4))           # 输出: 2j

3. 自定义复数类

class MyComplex:
    def __init__(self, real, imag):
        self.real = real
        self.imag = imag
    
    def __complex__(self):
        return complex(self.real, self.imag)

mc = MyComplex(3, 4)
print(complex(mc))             # 输出: (3+4j)

四、常见错误与陷阱

1. 字符串格式错误

complex("3 + 4j")  # 报错：ValueError（不能包含空格）
complex("3,4j")    # 报错：ValueError（逗号不是有效格式）

2. 混淆实数和复数运算

# 错误：实数与复数相加需显式转换
3 + complex(1, 2)  # 正确：(4+2j)
3 + (1+2j)        # 正确：(4+2j)
3 + "1+2j"        # 报错：TypeError

3. 精度问题

# 浮点数精度影响复数计算
z = complex(0.1, 0.2)
print(z)          # 输出: (0.1+0.2j)
print(z.real)     # 输出: 0.1（实际存储为近似值）

五、实战案例

1. 信号处理（傅里叶变换）

import numpy as np

# 生成复数信号
t = np.linspace(0, 1, 1000)
signal = np.exp(1j * 2 * np.pi * 5 * t)  # 5Hz 正弦波

# 计算频谱（简化示例）
spectrum = np.fft.fft(signal)
print(spectrum[5])  # 输出: 约 1000+0j（频率分量）

2. 电气工程（阻抗计算）

def calculate_impedance(r, l, c, frequency):
    """计算 RLC 电路的总阻抗"""
    omega = 2 * np.pi * frequency
    xl = 1j * omega * l      # 电感阻抗
    xc = -1j / (omega * c)   # 电容阻抗
    return r + xl + xc       # 总阻抗

# 示例：计算 50Hz 下的阻抗
impedance = calculate_impedance(100, 0.1, 1e-6, 50)
print(f"阻抗: {impedance:.2f} Ω")  # 输出: 阻抗: (100.00+31.39j) Ω

3. 复平面绘图

import matplotlib.pyplot as plt

# 生成复数集合（螺旋线）
points = [complex(i/10 * np.cos(i/5), i/10 * np.sin(i/5)) 
          for i in range(100)]

# 提取实部和虚部
real = [z.real for z in points]
imag = [z.imag for z in points]

# 绘制复平面
plt.figure(figsize=(6, 6))
plt.scatter(real, imag)
plt.axhline(y=0, color='k', linestyle='-')
plt.axvline(x=0, color='k', linestyle='-')
plt.title('Complex Plane')
plt.xlabel('Real')
plt.ylabel('Imaginary')
plt.grid(True)
plt.show()

六、最佳实践

1. 使用 cmath 而非 math：

   # 错误：math 模块不支持复数
   import math
   math.sqrt(-1)  # 报错：ValueError
   
   # 正确：使用 cmath
   import cmath
   cmath.sqrt(-1)  # 输出: 1j
   

2. 字符串转换时的验证：

   def parse_complex(s):
       try:
           return complex(s.replace(' ', ''))  # 移除空格
       except ValueError:
           return None
   print (parse_complex ("3 + 4j")) # 输出: (3+4j)
   

3.  复数的格式化输出：

   z = complex(3, 4)
   print(f"{z:.2f}")         # 输出: (3.00+4.00j)
   print(f"|z| = {abs(z):.2f}")  # 输出: |z| = 5.00

七、总结

complex() 函数是 Python 处理复数的基础工具，适用于科学计算、工程模拟等领域。需注意字符串格式要求和浮点数精度问题，推荐使用 cmath 模块处理复数的数学运算。复数的可视化（如复平面绘图）能帮助理解其几何意义。