How to take a subset of a list in python?

Question

What I'd like is this:

>>> [i if i!= 0 for i in [0,1,2,3]]
[1,2,3]

just like

>>> [i for i in [1,2,3,4]]
[1,2,3,4]

What's the simple solution that doesn't yield a syntax error?

Edit: assuming I don't want to use a for loop and appending all elements to a new list.

if comes later: `[i for i in [0,1,2,3] if i!= 0 ]` – ayhan Apr 29 '17 at 20:54 — ayhan, Apr 29 '17 at 20:54

score 3 · Accepted Answer · answered Apr 29 '17 at 21:02

3

use [i for i in [0,1,2,3] if i!=0] to get

[1, 2, 3]

answered Apr 29 '17 at 21:02

Vidur Khanna

score 1 · Answer 2 · answered Apr 29 '17 at 20:54

1

You can add if at the end:

[i for i in [0,1,2,3] if i!= 0]

answered Apr 29 '17 at 20:54

Spencer Wieczorek

score 1 · Answer 3 · answered Apr 29 '17 at 20:55

1

Just put the if i != 0 at the end of the list comprehension, like this:

[i for i in [0,1,2,3] if i!=0]

answered Apr 29 '17 at 20:55

Anonymous1847

score 0 · Answer 4 · answered Apr 29 '17 at 20:55

0

[i for i in [0,1,2,3] if 1 != 0]

answered Apr 29 '17 at 20:55

Matei

4 Answers4