List condition to get true or false on a new list

Question

I have a list:

>>> labels = ['setosa', 'setosa', 'versicolor', 'versicolor', 'virginica']

I want to create a new list with the same number of elements from True

to 'setosa'

index and False

elsewhere.

I tried like this

>>> b = 'setosa' in labels
>>> b
True

I need a list of 5 elements:

[True, True, False, False, False]

+3

Richard Rublev Apr 27. 17 at 11:37

3 answers

You can use , which in Python 2 returns : map

list

>>> labels = ['setosa', 'setosa', 'versicolor', 'versicolor', 'virginica']
>>> map('setosa'.__eq__, labels)
[True, True, False, False, False]

In Python 3, if you want a list:

>>> list(map('setosa'.__eq__, labels))
[True, True, False, False, False]

+2

Peter wood Apr 27. 17 at 11:48

you can use the function .append()

:

new_list = []

for element in labels: new_list.append('setosa' == element)

0

Veltro Apr 27. 17 at 11:40

Eugene yarmash · Accepted Answer · 2017-04-27T11:40:27+0000

Just use a list comprehension:

lst = [label == "setosa" for label in labels]