Lambda，包括if ... elif ... else

Question

我想在lambda函数中使用if … elif … else将lambda函数应用于DataFrame列。

df和代码都是smth。喜欢：

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)

显然，这种方式是行不通的。如果.... elif ....其他不适合lambda，有什么方法可以应用？如何使用List
Comprehension关联相同的结果？

感谢您的任何回复。

Answer 1

巢号if .. else：

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