python - Lambda including if...elif...else

Question

I want to apply a lambda function to a DataFrame column using if...elif...else within the lambda function.

The df and the code are smth. like:

df=pd.DataFrame({"one":[1,2,3,4,5],"two":[6,7,8,9,10]})

df["one"].apply(lambda x: x*10 if x<2 elif x<4 x**2 else x+10)

obviously this way it is not working. Is there a way to apply if....elif....else to lambda? How can I relize the same result with List Comprehension?

Thanks for any response.

Answer 1

Nest if .. elses:

lambda x: x*10 if x<2 else (x**2 if x<4 else x+10)