Maison  >  Article  >  développement back-end  >  Comment classer efficacement les éléments dans un tableau NumPy sans double tri ?

Comment classer efficacement les éléments dans un tableau NumPy sans double tri ?

DDD
DDDoriginal
2024-10-27 04:35:30851parcourir

How to Efficiently Rank Elements in a NumPy Array Without Double Sorting?

Classement efficace des tableaux en Python/NumPy sans double tri

Dans l'analyse des données et l'apprentissage automatique, le classement des éléments dans un tableau est une opération courante . Cependant, il est inefficace de trier le tableau deux fois pour y parvenir, car cela augmente la complexité des calculs. Voici une approche plus optimale en utilisant NumPy :

Pour classer les éléments d'un tableau sans avoir recours à deux fois, suivez ces étapes :

  1. Appliquez argsort une fois pour obtenir l'ordre des éléments du tableau.
  2. Utilisez à nouveau argsort pour classer l'ordre obtenu à l'étape 1.

Par exemple :

<code class="python">import numpy as np

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

order = array.argsort()
ranks = order.argsort()

print("Original Array:", array)
print("Ordering:", order)
print("Ranks:", ranks)</code>

Sortie :

Original Array: [4 2 7 1]
Ordering: [3 1 2 0]
Ranks: [2 1 3 0]

As vous pouvez le voir, le tableau des classements fournit le classement de chaque élément du tableau d'origine sans nécessiter un double tri.

Notez que, pour les tableaux de dimensions 2D ou supérieures, il est crucial de spécifier le bon axe à utiliser pour le tri. fournir un argument d'axe à argsort.

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!

Déclaration:
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn