Wie normalisiert man ein NumPy-Array auf einen Einheitsvektor?

Um einen NumPy-Array auf einen Einheitsvektor zu normalisieren, können Sie die numpy.linalg.norm Funktion verwenden, um die Größe des Vektors zu berechnen, und den Array anschließend durch diese Größe teilen. Hier ist ein Beispielcode-Schnipsel:

import numpy as np

# Initialize an array
arr = np.array([1, 2, 3])

# Calculate the magnitude of the vector
magnitude = np.linalg.norm(arr)

# Normalize the array by dividing by the magnitude
unit_vec = arr / magnitude

print(unit_vec)

Das gibt [0.26726124 0.53452248 0.80178373] aus.