Korrekter Python Weg zu: Arbeit am Original in for Schleife

Question

Ich habe ein ndarray 'n_data' von Schwimmern mit Form (N, 3). Ich möchte Änderungen an 'n_data' vornehmen (keine Kopie). Ich habe den folgenden Code Hexe funktioniert, aber es fühlt sich sehr schmutzig:

i = 0 
    for point in n_data: 
     point -= center 

     r = np.linalg.norm(point) 
     n_data[i] = point * (r ** gamma/r) 

     i += 1 

Ich versuchte np.nditer verwenden, aber für jede Iteration wäre es eine einzige Nummer und keine ndarray mit Form (3,), wie oben zu erhalten .

 for point in np.nditer(n_data, op_flags=['readwrite']): 

     point -= center 

     r = np.linalg.norm(point) 
     point = point * (r ** gamma/r) 

Vielen Dank im Voraus für alle Hilfe!

Answer 1

Eine vektorisierte Version der Übersetzung NumPy Begriffe und sieht immer noch nahe genug, um den ursprünglichen Code mit unter Verwendung von NumPy broadcasting und mit den optionalen Argumente axis Parameter mit np.linalg.norm, wie so wäre -

shifted = n_data - center 
R = np.linalg.norm(shifted, axis=1, keepdims=1) 
n_data = shifted * (R ** gamma/R) 

Hoffe, das fühlt sich weniger schmutzig!