我想在 lambda 函数中使用 if...elif...else 将 lambda 函数应用于 DataFrame 列。
df 和代码类似于：
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)
显然，这是行不通的。有没有办法将 if....elif....else 应用于 lambda？ 如何使用 List Comprehension 获得相同的结果？
【问题讨论】：
您可以使用多个 loc 运算符来完成。这是一个新创建的列，标记为“new”，并应用了条件计算：
df.loc[(df['one'] < 2), 'new'] = df['one'] * 10
df.loc[(df['one'] < 4), 'new'] = df['one'] ** 2
df.loc[(df['one'] >= 4), 'new'] = df['one'] + 10
【讨论】：
我不建议在这里使用apply：如果有更好的选择，应该避免使用。
例如，如果您对 Series 执行以下操作：
if cond1:
exp1
elif cond2:
exp2
else:
exp3
这通常是 np.where 或 np.select 的一个很好的用例。
numpy.where上面的ifelse链可以用
np.where(cond1, exp1, np.where(cond2, exp2, ...))
np.where 允许嵌套。通过一级嵌套，您的问题可以解决，
df['three'] = (
np.where(
df['one'] < 2,
df['one'] * 10,
np.where(df['one'] < 4, df['one'] ** 2, df['one'] + 10))
df
one two three
0 1 6 10
1 2 7 4
2 3 8 9
3 4 9 14
4 5 10 15
numpy.select允许灵活的语法并且易于扩展。它遵循形式，
np.select([cond1, cond2, ...], [exp1, exp2, ...])
或者，在这种情况下，
np.select([cond1, cond2], [exp1, exp2], default=exp3)
df['three'] = (
np.select(
condlist=[df['one'] < 2, df['one'] < 4],
choicelist=[df['one'] * 10, df['one'] ** 2],
default=df['one'] + 10))
df
one two three
0 1 6 10
1 2 7 4
2 3 8 9
3 4 9 14
4 5 10 15
and/or（类似于if/else）类似于if-else，需要lambda：
df['three'] = df["one"].apply(
lambda x: (x < 2 and x * 10) or (x < 4 and x ** 2) or x + 10)
df
one two three
0 1 6 10
1 2 7 4
2 3 8 9
3 4 9 14
4 5 10 15
比apply快仍然快的循环解决方案。
df['three'] = [x*10 if x<2 else (x**2 if x<4 else x+10) for x in df['one']]
# df['three'] = [
# (x < 2 and x * 10) or (x < 4 and x ** 2) or x + 10) for x in df['one']
# ]
df
one two three
0 1 6 10
1 2 7 4
2 3 8 9
3 4 9 14
4 5 10 15
【讨论】：
defafunction
df["one"]=df["one"].apply(lambda x: x**2 if x==2 else lambda2(x)) lambda2= lambda x: x**3 if x==3 else lambda3(x) lambda3=lambda x: x**4 if x==4 else lambda4(x) lambda4=lambda x: x**5 if x==5 else x
为了可读性，我更喜欢编写一个函数，尤其是当您处理许多条件时。对于原始问题：
def parse_values(x):
if x < 2:
return x * 10
elif x < 4:
return x ** 2
else:
return x + 10
df['one'].apply(parse_values)
【讨论】：
if-else 链变长时
巢if .. elses:
lambda x: x*10 if x<2 else (x**2 if x<4 else x+10)
【讨论】：
lambda 内部”，那么是的，这在任何地方都有效。事实上，这是我知道的唯一一种制作单行 if-elif-else 的方法，除了使用 or/and 技巧。
x==0 - 在x*10 之后的or 会向前滑动，而不是返回0
a or b 不返回 True 或 False，但如果计算结果为 a，则返回 a，否则 b。
True。