为什么python lambda看到的是一个系列而不是一个值？

Question

我创建了一个 Pandas DataFrame：

df = pd.DataFrame( {'some_number' : [1,2,3,4,5,6]})

然后我想添加一个名为 is_even 的列：

df.assign(
    is_even = lambda x : 'YES' if x.some_number % 2 == 0 else 'NO'
)

我收到一个错误：

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

我知道错误告诉我 if 语句之后的x.some_number是一个系列。 这让我感到困惑，因为如果我这样做：

df.assign(
    is_even = lambda x : 'YES' if 1==2 else x.some_number
)

它可以工作并生成以下输出：

这表明 x.some_number 实际上不是一个系列，而是一个标量值。

我知道还有其他方法可以完成我想要完成的任务。 但我对这种行为很感兴趣。

为什么，当 x.some_number 在if子句之后时，它被视为一个系列，但当它在else子句中使用时，它被视为一个值？

INSTALLED VERSIONS
------------------
python           : 3.8.0.final.0
python-bits      : 32
OS               : Windows
OS-release       : 10
machine          : AMD64
processor        : Intel64 Family 6 Model 158 Stepping 10, GenuineIntel
byteorder        : little
LOCALE           : English_United States.1252

pandas           : 0.25.3
numpy            : 1.17.4
IPython          : 7.10.0
matplotlib       : 3.1.2

Answer 1

问题仅在于 if 语句，您在第一个示例中将序列与标量值进行比较。 这永远不会奏效。 第二个示例有效，因为您有一个标量 if 语句（当然可以）并且您返回一个系列。 返回一个系列（或一个标量）正是传递给assign的函数需要做的。

现在，您真正想做的是逐行比较。 使用apply

df['is_even'] = df.some_number.apply(lambda x: 'YES' if x % 2 == 0 else 'NO' )

这里，x 是一个标量，if 语句按预期工作。 或者，您可以结合分配和 lambda 函数

df.assign(
    is_even = lambda x : x.some_number.apply(lambda x: 'YES' if x % 2 == 0 else 'NO')
)

再次注意第一个示例的不同之处：外部 lambda 确保内部 lambda 只需要处理if x % 2 == 0标量。 外部 lambda 返回一个系列，就像在你的第二个例子中一样。

Answer 2

你的证明不成立。 Pandas Dataframe.assign似乎能够处理系列或标量并将其应用于数据帧。

In [7]: df.assign(is_even=lambda x: x.some_number[0] )                                                                 
Out[7]: 
   some_number  is_even
0            1        1
1            2        1
2            3        1
3            4        1
4            5        1
5            6        1

如果您仔细阅读文档，您将看到参数接受回调或系列并根据类型应用它。

列名是关键字。 如果这些值是可调用的，它们将在 DataFrame 上计算并分配给新列。 可调用对象不得更改输入数据帧（尽管熊猫不检查它）。 如果值不可调用（例如系列、标量或数组），则只需对其进行赋值。

另外，如果你深入研究一下源代码：

# >= 3.6 preserve order of kwargs
if PY36:
for k, v in kwargs.items():
    data[k] = com.apply_if_callable(v, data)

您可以查看它是否是可调用的，它将整个数据帧传递给您的可调用对象。

Answer 3

df.loc[df.some_number % 2 ==0,'is_even'] = 'Yes'
df.loc[df.some_number % 2 !=0,'is_even'] = 'No'
print(df)

或使用申请

df['is_even'] = df.some_number.apply(lambda x: 'YES' if x % 2 == 0 else 'NO' )