Comment trouver la plus petite différence entre un nombre donné et chaque élément d'une liste en Python?

Question

Dire que j'ai:

[1, 2, 3, 4]

et l'entier

6

Je veux comparer 6 avec tous les éléments dans la liste et retourner l'élément avec la plus petite différence de valeur absolue qui ce cas est 4. Y a-t-il un moyen efficace de le faire?

Answer 1

Vous pouvez utiliser argmin sur la différence absolue pour extraire l'indice, qui peut ensuite être utilisé pour extraire l'élément:

a = np.array([1, 2, 3, 4]) 

a[np.abs(a - 6).argmin()] 
# 4