Wie bekomme ich eine Liste von Indizes von Nicht-Null-Elementen in einer Liste?

Ich habe eine Liste, die enthält immer nur Einsen und Nullen.
Ich brauche eine Liste der nicht-null-Indizes der Liste:

a = [0, 1, 0, 1, 0, 0, 0, 0]
b = []
for i in range(len(a)):
    if a[i] == 1:  b.append(i)
print b

Was wäre der "pythonic" Art und Weise, dies zu erreichen ?

InformationsquelleAutor der Frage George Profenza | 2010-11-06

  1. 55
    [i for i, e in enumerate(a) if e != 0]

    InformationsquelleAutor der Antwort Ignacio Vazquez-Abrams

  2. 27

    Nicht wirklich eine "neue" Antwort, aber numpy hat diese eingebaut. 

    import numpy as np
a = [0, 1, 0, 1, 0, 0, 0, 0]
nonzeroind = np.nonzero(a)[0] # the return is a little funny so I use the [0]
print nonzeroind
[1 3]

    InformationsquelleAutor der Antwort Brian Larsen

  3. 6

    Seit THC4k erwähnt komprimieren (in python2.7+)

    >>> from itertools import compress, count
>>> x = [0, 1, 0, 1, 0, 0, 0, 0]
>>> compress(count(), x)
<itertools.compress object at 0x8c3666c>   
>>> list(_)
[1, 3]

    InformationsquelleAutor der Antwort John La Rooy

  4. 2

    Wünschte nur hinzufügen Erklärung für 'lustig' Ausgabe aus dem vorherigen asnwer. Ergebnis ist ein Tupel enthält, die Vektoren von Indizes für jede dimension der matrix. In diesem Fall Benutzer ist die Verarbeitung, was als ein Vektor in numpy, damit die Ausgabe ist ein Tupel mit einem element.

    import numpy as np
a = [0, 1, 0, 1, 0, 0, 0, 0]
nonzeroind = np.nonzero(a) 
print nonzeroind
(array([1, 3]),)

    InformationsquelleAutor der Antwort Lexa

Schreibe einen Kommentar