Convert boolean in Integer Lage in Python

Question

ich eine boolean Liste haben, sagen:

x = [True, False, False, True] 

Wie Sie diese Liste konvertieren Standorte in Integer, so dass Sie das Ergebnis:

y = [1, 4] 

?

Answer 1

Sie eine Liste Verständnis in Kombination mit der enumerate Funktion zum Beispiel verwenden:

>>> x = [True, False, False, True] 
>>> [index for index, element in enumerate(x, start=1) if element] 
[1, 4] 

Alternativ, wenn Sie bereit sind, NumPy zu verwenden und ein Ergebnis vom Typ numpy.ndarray, bekommt es gibt eine NumPy Funktion, (fast) macht was du brauchst: numpy.where.

>>> import numpy 
>>> numpy.where(x) 
(array([0, 3]),) 
>>> numpy.where(x)[0] + 1 
array([1, 4]) 

Die seltsamen [0] in der Zeile darüber ist da, weil numpy.where immer seine Ergebnisse in einem Tupel zurück: ein Element des Tupels für jede Dimension des Eingangsfeldes. Da in diesem Fall das Eingabearray eindimensional ist, interessieren wir uns nicht wirklich für die äußere Tupelstruktur, daher verwenden wir die Indizierungsoperation [0], um das tatsächliche Array herauszuziehen, das wir brauchen. Die + 1 gibt es von Python/NumPy Standard-0-basierte Indizierung auf die 1-basierte Indizierung, dass es aussieht, als ob Sie hier möchten.

Wenn Sie mit großen Eingabedaten arbeiten (und insbesondere wenn die Eingabeliste bereits in Form eines NumPy-Arrays vorliegt), ist die NumPy-Lösung wahrscheinlich wesentlich schneller als das Listenverständnis.

Answer 2

Ein einfacher Einzeiler wäre:

[i+1 for i in range(len(x)) if x[i]] 

Answer 3

Verwenden enumerate innerhalb einer Liste Verständnis:

>>> [i for i, j in enumerate(x, 1) if j] 
[1, 4]