带有 && python 的 Lambda 表达式

Question

我正在尝试使用 lambda 表达式来提取具有两个条件的日期。 当我写这篇文章时，它可以工作

newdf.loc[lambda newdf: newdf['yearly_premium'] > 2784, :]

但是当我添加另一个条件时它不会：/

newdf.loc[lambda newdf: newdf['yearly_premium'] > 2784 and newdf['Total_Claim_Amount'] < 1021.654, :]

谁能告诉我如何调整它？谢谢各位

Answer 1

将lambda 函数传递给DataFrame.loc 时，表达式必须计算为布尔数组。可以使用布尔运算符指定单个表达式中的多个条件，以便整个表达式的评估也是一个布尔数组，例如：

lambda newdf: (newdf['yearly_premium'] > 2784) & (newdf['Total_Claim_Amount'] < 1021.654)

请注意，每个条件都用括号括起来。这是因为默认的 Python 评估顺序不会产生您期望的结果。

示例

import pandas as pd

df = pd.DataFrame({'x': [1, 2, 3], 'y': [3, 4, 5]})
newdf = df.loc[lambda df: (df['x'] > 1) & (df['y'] < 5)]

print(df)
print(newdf)

输出

   x  y
0  1  3
1  2  4
2  3  5

   x  y
1  2  4

关于使用lambda的注意事项

值得注意的是，在此特定用例中不需要 lambda 函数。在前面的例子中，直接使用标签也可以达到同样的效果，例如：

newdf = df.loc[(df['x'] > 1) & (df['y'] < 5)]

有关其他文档，请参阅 Indexing and selecting data。