NumPy: Erstes Element in sortiertem Array finden, das größer als ein gegebener Skalar ist

Siehe auch How to find first element in sorted numpy array that is smaller than a given scalar

Du kannst np.searchsorted() so verwenden:

find_first_larger.py
idx = np.searchsorted(arr, scalar, side='right') + 1

Vollständiges Beispiel:

find_first_larger_example.py
import numpy as np

# Beispiel-Array
arr = np.array([1, 2, 3, 4, 5])

# Wir suchen nach diesem Wert
scalar = 3.5

# numpy.searchsorted() verwenden, um das erste Element im Array zu finden, das größer als der Skalar ist
# Wir müssen + 1 verwenden, da searchsorted() das erste Element zurückgibt, das kleiner als der Skalar ist
idx = np.searchsorted(arr, scalar, side='right') + 1

# Den resultierenden Index ausgeben
print(idx)

Check out similar posts by category: Python