Comment obtenir une liste d’indices d’éléments non nuls dans une liste ?

Question

J’ai une liste qui ne contiendra toujours que des uns et des zéros. J’ai besoin d’obtenir une liste des indices non nuls de la liste :

Quelle serait la manière « pythonique » d’y parvenir ?

Answer 1

<pre><code></code></pre>

Answer 2

Pas vraiment une « nouvelle » réponse , mais numpy a cela intégré aussi.

Answer 3

Depuis que THC4k a mentionné compress (disponible en python2.7+)

Answer 4

Je voulais juste ajouter une explication pour la sortie « drôle » de l’asnwer précédent. Le résultat est un tuple qui contient des vecteurs d’index pour chaque dimension de la matrice. Dans ce cas, l’utilisateur traite ce qui est considéré comme un vecteur en numpy, de sorte que la sortie est tuple avec un élément.

Answer 5

Comparaison temporelle des deux réponses w.r.t longueur de la liste

Avec une longueur de liste de 100000, en modifiant le nombre de ceux dans la liste

Le nombre de ceux-ci n’affecte que la première option. np.nonzero() est meilleur avec une grande quantité d’éléments non nuls. Si la longueur est inférieure à 10000, la première option est plus rapide.