One-line list comprehension: if-else variants

PythonList ComprehensionTernary OperatorConditional Operator

Python Problem Overview


It's more about python list comprehension syntax. I've got a list comprehension that produces list of odd numbers of a given range:

[x for x in range(1, 10) if x % 2]

This makes a filter - I've got a source list, where I remove even numbers (if x % 2). I'd like to use something like if-then-else here. Following code fails:

>>> [x for x in range(1, 10) if x % 2 else x * 100]
  File "<stdin>", line 1
    [x for x in range(1, 10) if x % 2 else x * 100]
                                         ^
SyntaxError: invalid syntax

There is a python expression like if-else:

1 if 0 is 0 else 3

How to use it inside a list comprehension?

Python Solutions


Solution 1 - Python

x if y else z is the syntax for the expression you're returning for each element. Thus you need:

[ x if x%2 else x*100 for x in range(1, 10) ]

The confusion arises from the fact you're using a filter in the first example, but not in the second. In the second example you're only mapping each value to another, using a ternary-operator expression.

With a filter, you need:

[ EXP for x in seq if COND ]

Without a filter you need:

[ EXP for x in seq ]

and in your second example, the expression is a "complex" one, which happens to involve an if-else.

Solution 2 - Python

[x if x % 2 else x * 100 for x in range(1, 10) ]

Solution 3 - Python

You can do that with list comprehension too:

A=[[x*100, x][x % 2 != 0] for x in range(1,11)]
print A

Solution 4 - Python

Just another solution, hope some one may like it :

Using: [False, True][Expression]

>>> map(lambda x: [x*100, x][x % 2 != 0], range(1,10))
[1, 200, 3, 400, 5, 600, 7, 800, 9]
>>>

Solution 5 - Python

#how you can squre a list of an array of negative and positive values

my_list=[-2,-3,0,1,5]
squred_values=[]
  

squred_values=[-i**2 if i<0 else i**2 for i in my_list]

#or

for i in my_list:
    if i<0:
        squred_values.append( -i**2)`enter code here`
    else:
        squred_values.append( i**2) 

Solution 6 - Python

two for in one list comprehension

ex = [['obi', 'is', '#alive'],['oge', 'is', 'beautiful'],
                ['Ade', 'the', '#comedian', 'de', '#rich'],['Jesus', 'wept']]
res = [j if j.startswith("#") else "_"+j for i in ex for j in i]

Attributions

All content for this solution is sourced from the original question on Stackoverflow.

The content on this page is licensed under the Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) license.

Content TypeOriginal AuthorOriginal Content on Stackoverflow
QuestionducinView Question on Stackoverflow
Solution 1 - Pythonshx2View Answer on Stackoverflow
Solution 2 - PythonlucasgView Answer on Stackoverflow
Solution 3 - PythonStefan GruenwaldView Answer on Stackoverflow
Solution 4 - PythonJamesView Answer on Stackoverflow
Solution 5 - PythonMidoView Answer on Stackoverflow
Solution 6 - PythonJagat DeshmukhView Answer on Stackoverflow