Comment trouver la valeur la plus proche dans un tableau NumPy ?

Trouver la valeur la plus proche dans un tableau Numpy

Dans numpy, trouver la valeur la plus proche d'une valeur cible donnée peut être obtenu grâce à un processus simple . Définissons une fonction pour répondre à cette exigence :

import numpy as np

def find_nearest(array, value):
    array = np.asarray(array)
    idx = (np.abs(array - value)).argmin()
    return array[idx]

Cette fonction accepte un tableau et une valeur cible. Il convertit le tableau en tableau numpy et calcule les différences absolues entre chaque élément du tableau et la valeur cible. L'index de l'élément avec la plus petite différence absolue est déterminé à l'aide de argmin(). Enfin, l'élément à cet index est renvoyé.

Exemple d'utilisation :

Illustrons l'utilisation de la fonction find_nearest() avec un exemple :

array = np.random.random(10)
print(array)

print(find_nearest(array, value=0.5))

Dans cet exemple, un tableau aléatoire de 10 éléments est créé. La fonction find_nearest() est ensuite utilisée pour rechercher l'élément du tableau le plus proche de 0,5. La sortie sera la valeur la plus proche du tableau à 0,5.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!