Question

我对lambda函数和标量有dict理解：

d = {k: lambda x : x.sum() if 'a' in k else 'yes' for k in ['bac','sss','asa']}
print (d)
{'bac': <function <dictcomp>.<lambda> at 0x00000000031891E0>, 
 'sss': <function <dictcomp>.<lambda> at 0x000000000D887EA0>, 
 'asa': <function <dictcomp>.<lambda> at 0x000000000D887B70>}

如果同时需要两个标量，则效果很好：

d = {k: 'no' if 'a' in k else 'yes' for k in ['bac','sss','asa']}
print (d)
{'bac': 'no', 'sss': 'yes', 'asa': 'no'}

预期输出-标量和lambda函数的组合：

print (d)
{'bac': <function <dictcomp>.<lambda> at 0x00000000031891E0>, 
 'sss': 'yes', 
 'asa': <function <dictcomp>.<lambda> at 0x000000000D887B70>}

发生了什么事？为什么不起作用？正确的方法是什么？

Answer 1

您的语法解析如下（注意括号的位置）：

{k: lambda x : (x.sum() if 'a' in k else 'yes') for k in ['bac','sss','asa']}
#              ^                              ^

您要

{k: (lambda x : x.sum()) if 'a' in k else 'yes' for k in ['bac','sss','asa']}
#   ^                  ^

这是因为lambda比if-else具有lower precedence。

一个更简单的示例也说明了这一点：

>>> lambda x: 0 if False else True
<function <lambda> at 0x7efdbe55abf8>
>>> lambda x: (0 if False else True)
<function <lambda> at 0x7efdbe55ac80>
>>> (lambda x: 0) if False else True
True

Answer 2

d = {k: (lambda x : x.sum()) if 'a' in k else 'yes' for k in ['bac','sss','asa']}

应该工作。原因（据我了解）是，解析lambda的边界与您的理解相重叠-基本上，python将lambda解析为x.sum() if 'a' in k else 'yes' –参照本地人

带Lambda函数和标量的Dict理解

2 个答案: