if, and, elif 都包含在 apply(lambda x:

Question

我有以下代码：

data['ShortLongFlag'] = data['End DateTime'].apply(lambda x:
                                                   -1 if (x.month == 3 and x.date() in shortlongdates == True) else (1 if (x.month == 10 and x.date() in shortlongdates == True) else 0))

我想要做的是：

在我的数据框中创建一个新列，根据以下条件填充 -1、0 或 1：

• -1 如果我的 datetime 列中的值的月份等于 3 并且日期在我的名为“shortlongdates”的日期列表中
• 1 如果我的日期时间列中的值的月份等于 10 并且日期在我的名为“shortlongdates”的日期列表中
• 0 否则

现在所有值都在新列中输出为 0...为什么？

Answer 1

这个问题的原因是chaining comparison operators

比较可以任意链接，例如，x

比较，包括成员资格测试和身份测试have same precedence。

即

x.month == 3 and x.date() in shortlongdates == True

与

相同

x.month == 3 and x.date() in shortlongdates and shortlongdates == True

注意可以写成x.month == 3 and x.date() in shortlongdates，也可以用括号。然而，正如 cmets 中所述，这个 lambda 最好写成常规函数。

def replace_value(x):
    if x.date() in shortlongdates:
        return {3:-1, 10:1}.get(x.month, 0)
    return 0

如果您坚持，我会留给您将其转换回 [更简单的] lambda。

Answer 2

所以我们有这个不可读的东西：

lambda x: -1 if (x.month == 3 and x.date() in shortlongdates == True) else (1 if (x.month == 10 and x.date() in shortlongdates == True) else 0)

让我们把它写成一个标准函数：

def filterfn(x):
    if x.month == 3 and x.date() in shortlongdates:
        return -1
    elif x.month == 10 and x.date() in shortlongdates:
        return 1
    else:
        return 0

删除一些陌生的测试（冗余 == True）和括号。

x.date() 真的是可调用的吗？应该是x.date？ 如果没有看到您的data 和listofdates，就不可能说这会如何失败，但至少我们可以看到它在做什么。