如何在NumPy中生成复平面上均匀分布的二维复数数组？

要生成你需要的二维复数数组，核心是先构建实部和虚部的网格，再将两者组合为复数类型的数组。以下是两种简洁的实现方式：

方法一：使用np.mgrid直接构建 #

np.mgrid可以生成多维网格，我们先分别生成实部（x轴）和虚部（y轴）的网格，再通过x + 1j*y组合成复数数组：

import numpy as np

# 生成二维网格：实部从-1到1，虚部从-1到1，每个维度取20个点
X, Y = np.mgrid[-1:1:20j, -1:1:20j]  # 20j表示取20个点，而非指定步长

# 组合为复数数组，dtype自动为complex类型
complex_grid = X + 1j * Y

print(complex_grid.shape)  # 输出 (20, 20)，对应20x20的网格
print(complex_grid.dtype)  # 输出 complex128

方法二：使用np.meshgrid构建 #

如果更习惯meshgrid的逻辑，也可以用它生成网格后组合：

import numpy as np

# 生成一维实部和虚部数组
x = np.linspace(-1, 1, 20)
y = np.linspace(-1, 1, 20)

# 生成网格矩阵
X, Y = np.meshgrid(x, y)

# 组合为复数数组
complex_grid = X + 1j * Y

关键说明 #

• 两种方法生成的数组形状都是(20,20)，每个元素对应复平面上的一个点x+yi，完全覆盖你指定的矩形区域。
• X + 1j*Y是将实数网格转换为复数数组的核心操作，NumPy会自动处理类型转换，最终数组的dtype为complex128，能直接用于后续复数运算。
• 若要调整区域范围，只需修改linspace或mgrid中的起止值即可，比如把实部范围改成0到2、虚部改成1到3，逻辑完全一致。

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