NumPy: Array-Index finden, der einem Wert am nächsten liegt

Nehmen wir an, du hast ein 1D-Array wie

numpy_argmin_closest.py
arr = np.linspace(0, 10, 100)

und möchtest den Array-Index finden, bei dem der Wert am nächsten an 8.5 liegt.

Du kannst dies tun, indem du zuerst die absolute Differenz zu 8.5 berechnest:

numpy_abs_diff.py
np.abs(arr - 8.5)

und nun np.argmin verwendest, um den Array-Index zu finden, bei dem der Wert minimal ist (d.h. den Index, bei dem der Wert von arr am nächsten an 8.5 liegt)

numpy_argmin_example.py
np.argmin(np.abs(arr - 8.5))

Check out similar posts by category: Python