Selectare Condițională Ramificată cu np.select în NumPy

În timp ce np.where() permite doar o singură condiție booleană binară (dacă/altfel), funcția np.select() evaluează o listă întreagă de condiții în ordine, returnând valoarea corespunzătoare primei condiții adevărate. Este echivalentul vectorizat complet al structurilor if-elif-else din Python.

import numpy as np

note = np.array([4, 6, 8, 10, 5])

conditii = [
    note >= 9,
    note >= 7,
    note >= 5
]
rezultate = ["EXCELENT", "BUN", "PROMOVAT"]

calificative = np.select(conditii, rezultate, default="RESPINS")
print(calificative)