2D阵列创建包含沿每个轴的实数和虚数

Question

这是一个简单的问题，尽管我很难在网上找到答案，主要是因为不确定最佳搜索条件。

基本上，我想创建一个二维数组，比如：

[0 + 0j, 1+0j, 2+0j
 0 + 1j, 1+1j, 2+1j
 0 + 2j, 1+2j, 2+2j]

我可以用两个嵌套循环来实现这一点，尽管我觉得必须有一个使用numpy和数组的1行解决方案？

非常感谢

Answer 1

>>> a = np.arange(3)
>>> a + a[:,np.newaxis] * 1j
array([[ 0.+0.j,  1.+0.j,  2.+0.j],
       [ 0.+1.j,  1.+1.j,  2.+1.j],
       [ 0.+2.j,  1.+2.j,  2.+2.j]])

Answer 2

您可以使用嵌套列表综合：

[[a+b*1j for a in range(3)] for b in range(3)]