Python list comprehension for if-else statements

Question

Is there a way to express the following code through Python list comprehension?

newlist = []
for i,j in enumerate(old_list):
if i==0:
    newlist.append(j*2+1)
else:
    newlist.append(j*2)

Use a conditional expression: `[j*2+1 if not i else j*2 for i, j in enumerate(old_list)]` — Martijn Pieters, Apr 11 '16 at 08:59

score 2 · Accepted Answer · answered Apr 11 '16 at 08:58

2

Sure, why not?

[j*2+1 if i==0 else j*2 for i, j in enumerate(old_list)]

answered Apr 11 '16 at 08:58

Avinash Raj

score 1 · Answer 2 · answered Apr 11 '16 at 08:59

1

I would do it with an inline if, like this:

newlist = [j * 2 + (1 if i == 0 else 0) for i, j in enumerate(old_list)]

answered Apr 11 '16 at 08:59

2 Answers2