Maison développement back-end Tutoriel Python Comment sélectionner efficacement des index de colonnes spécifiques par ligne dans NumPy ?

Comment sélectionner efficacement des index de colonnes spécifiques par ligne dans NumPy ?

Oct 27, 2024 am 12:32 AM

How to Efficiently Select Specific Column Indexes per Row in NumPy?

NumPy : sélection efficace d'index de colonnes spécifiques par ligne

La sélection des données est une opération cruciale dans l'analyse des données. Lorsque vous travaillez avec des tableaux NumPy, la sélection de colonnes spécifiques de chaque ligne peut être une tâche courante. Cette sélection peut être effectuée avec différentes méthodes, mais la sélection de colonnes basée sur une liste d'index par ligne nécessite une approche plus efficace.

Utilisation de tableaux booléens pour la sélection directe

Si vous disposez d'un tableau booléen indiquant les colonnes à sélectionner, vous pouvez utiliser la sélection directe pour extraire efficacement les valeurs souhaitées. Des tableaux booléens peuvent être créés en comparant une liste d'index avec la plage de colonnes. Par exemple, étant donné une matrice X et une liste d'index Y comme décrit dans la question, vous pouvez créer un tableau booléen b comme suit :

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

X = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
Y = np.array([1, 0, 2])
b = np.arange(X.shape[1])[np.isin(np.arange(X.shape[1]), Y)]</code>
Copier après la connexion

Avec le tableau booléen b, une sélection directe peut être effectuée :

<code class="python">result = X[np.arange(X.shape[0]), b]</code>
Copier après la connexion

Cette méthode fournit un moyen rapide de sélectionner des colonnes spécifiques en fonction du tableau booléen.

Méthodes alternatives

Vous pouvez également utiliser np.arange et sélection directe basée sur la liste d'index Y. Cette approche consiste à créer un tableau d'indices et à sélectionner dans la matrice X en conséquence :

<code class="python">result = X[np.arange(X.shape[0]), Y]</code>
Copier après la connexion

Conclusion

La sélection d'index de colonnes spécifiques par ligne dans NumPy peut être effectuée efficacement à l'aide de tableaux booléens. Cette méthode permet une sélection rapide et simple de colonnes basée sur une liste d'index. Pour de larges tableaux de données, cette approche offrira des avantages significatifs en termes de performances par rapport aux méthodes basées sur les itérations.

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 de ce site Web
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

Article chaud

Combien de temps faut-il pour battre Split Fiction?
3 Il y a quelques semaines By DDD
Repo: Comment relancer ses coéquipiers
3 Il y a quelques semaines By 尊渡假赌尊渡假赌尊渡假赌
Hello Kitty Island Adventure: Comment obtenir des graines géantes
3 Il y a quelques semaines By 尊渡假赌尊渡假赌尊渡假赌
Musée à deux points: toutes les expositions et où les trouver
3 Il y a quelques semaines By 尊渡假赌尊渡假赌尊渡假赌

Article chaud

Combien de temps faut-il pour battre Split Fiction?
3 Il y a quelques semaines By DDD
Repo: Comment relancer ses coéquipiers
3 Il y a quelques semaines By 尊渡假赌尊渡假赌尊渡假赌
Hello Kitty Island Adventure: Comment obtenir des graines géantes
3 Il y a quelques semaines By 尊渡假赌尊渡假赌尊渡假赌
Musée à deux points: toutes les expositions et où les trouver
3 Il y a quelques semaines By 尊渡假赌尊渡假赌尊渡假赌

Tags d'article chaud

Bloc-notes++7.3.1

Bloc-notes++7.3.1

Éditeur de code facile à utiliser et gratuit

SublimeText3 version chinoise

SublimeText3 version chinoise

Version chinoise, très simple à utiliser

Envoyer Studio 13.0.1

Envoyer Studio 13.0.1

Puissant environnement de développement intégré PHP

Dreamweaver CS6

Dreamweaver CS6

Outils de développement Web visuel

SublimeText3 version Mac

SublimeText3 version Mac

Logiciel d'édition de code au niveau de Dieu (SublimeText3)

Comment utiliser la belle soupe pour analyser HTML? Comment utiliser la belle soupe pour analyser HTML? Mar 10, 2025 pm 06:54 PM

Comment utiliser la belle soupe pour analyser HTML?

Filtrage d'image en python Filtrage d'image en python Mar 03, 2025 am 09:44 AM

Filtrage d'image en python

Comment télécharger des fichiers dans Python Comment télécharger des fichiers dans Python Mar 01, 2025 am 10:03 AM

Comment télécharger des fichiers dans Python

Comment utiliser Python pour trouver la distribution ZIPF d'un fichier texte Comment utiliser Python pour trouver la distribution ZIPF d'un fichier texte Mar 05, 2025 am 09:58 AM

Comment utiliser Python pour trouver la distribution ZIPF d'un fichier texte

Comment travailler avec des documents PDF à l'aide de Python Comment travailler avec des documents PDF à l'aide de Python Mar 02, 2025 am 09:54 AM

Comment travailler avec des documents PDF à l'aide de Python

Comment se cacher en utilisant Redis dans les applications Django Comment se cacher en utilisant Redis dans les applications Django Mar 02, 2025 am 10:10 AM

Comment se cacher en utilisant Redis dans les applications Django

Comment effectuer l'apprentissage en profondeur avec TensorFlow ou Pytorch? Comment effectuer l'apprentissage en profondeur avec TensorFlow ou Pytorch? Mar 10, 2025 pm 06:52 PM

Comment effectuer l'apprentissage en profondeur avec TensorFlow ou Pytorch?

Comment implémenter votre propre structure de données dans Python Comment implémenter votre propre structure de données dans Python Mar 03, 2025 am 09:28 AM

Comment implémenter votre propre structure de données dans Python

See all articles