沿每个轴包含实数和虚数的二维数组创建

Question

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

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

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

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

非常感谢

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)]