Generieren eine zweidimensionale Anordnung von komplexen Zahlen

Question

ich eine zweidimensionale Anordnung einer bestimmten Größe mit komplexen Zahlen wie in diesem Beispiel erzeugen möge:

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

Hier ist ein Weg, um diesen
mit np.indices() Eingriff zu erreichen mit 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) 

Ausgang:

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