Сформировать двумерный массив комплексных чисел

Question

Я хочу, чтобы создать двумерный массив заданного размера, содержащий комплексные числа, как в этом примере:

>>> generate_array((2, 3)) 
array([[ 0.+0.j, 1.+0.j, 2.+0.j], 
     [ 0.+1.j, 1.+1.j, 2.+1.j]]) 

Answer 1

Вот способ для достижения этой
используя np.indices() помолвлено с np.vectorize():

def generate_array(dim): 
    X, Y = np.indices(dim) 
    return np.array(np.vectorize(complex)(X, Y)) 

ar = generate_array((2, 3)) 
print(ar) 

Выход:

[[ 0.+0.j 0.+1.j 0.+2.j] 
[ 1.+0.j 1.+1.j 1.+2.j]] 

Answer 2

def generate_array(m, n): 
    return (np.arange(m) * 1j)[:, None] + np.arange(n) 


generate_array(2, 3) 
Out: 
array([[ 0.+0.j, 1.+0.j, 2.+0.j], 
     [ 0.+1.j, 1.+1.j, 2.+1.j]])