Comment puis-je trouver efficacement des lignes uniques dans un tableau NumPy ?

Trouver des lignes uniques dans un tableau NumPy : une solution efficace

Dans l'analyse et le traitement des données, il est souvent nécessaire d'extraire des valeurs uniques de un ensemble de données donné. Dans ce contexte, considérons le problème de la recherche de lignes uniques dans un tableau NumPy.

Objectif :

Étant donné un tableau NumPy, le but est d'identifier et d'obtenir un tableau contenant uniquement les lignes uniques du tableau d'origine.

Efficace Solution :

Depuis la version 1.13 de NumPy, une solution efficace pour rechercher des lignes uniques a été introduite. En tirant parti de la fonction np.unique et en spécifiant le paramètre axis, nous pouvons y parvenir facilement :

unique_rows = np.unique(original_array, axis=0)

En définissant le paramètre axis sur 0, nous demandons à NumPy d'analyser chaque ligne du tableau d'origine indépendamment. Cette opération compare les lignes élément par élément et renvoie un nouveau tableau qui contient uniquement les lignes uniques.

Exemple :

Considérez le tableau NumPy suivant a :

a = np.array([[1, 1, 1, 0, 0, 0],
               [0, 1, 1, 1, 0, 0],
               [0, 1, 1, 1, 0, 0],
               [1, 1, 1, 0, 0, 0],
               [1, 1, 1, 1, 1, 0]])

Pour obtenir les lignes uniques de a, nous pouvons utiliser ce qui suit code :

unique_rows = np.unique(a, axis=0)

Cela produira un nouveau tableau unique_rows qui contient les lignes suivantes :

unique_rows = np.array([[1, 1, 1, 0, 0, 0],
                         [0, 1, 1, 1, 0, 0],
                         [1, 1, 1, 1, 1, 0]])

Comme nous pouvons le voir, les lignes uniques du tableau d'origine ont été extraites avec succès.

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!