使用pandas编写lambda函数触发SyntaxError语法错误的原因是什么

报错原因 #

你代码的语法错误来自不完整的嵌套三元表达式：
Python 三元表达式的强制语法规则为 返回值1 if 判断条件 else 返回值2，三个部分缺一不可，不允许仅写if分支不写最终兜底的else分支。
你现有代码的内层判断只写了 'down' if data['Rolling'].shift(1) > data['Rolling']，没有覆盖data['Rolling'].shift(1) == data['Rolling']的相等场景，缺少对应的返回值，因此触发SyntaxError。

修正方案 #

补充内层判断的else分支即可，比如相等时返回flat（可根据你的业务需求修改为对应值）：

# 补全三元表达式的所有分支
data['Direction'] = lambda x : 'up' if data['Rolling'].shift(1) < data['Rolling'] else ('down' if data['Rolling'].shift(1) > data['Rolling'] else 'flat')

额外提示：你当前的写法会把lambda函数直接赋值给data['Direction']列，而非执行后的计算结果。如果是要对pandas DataFrame的整列做值映射，更建议用numpy.where的写法，性能更高且逻辑更清晰：

import numpy as np
data['Direction'] = np.where(data['Rolling'].shift(1) < data['Rolling'], 'up',
                             np.where(data['Rolling'].shift(1) > data['Rolling'], 'down', 'flat'))

内容的提问来源于stack exchange，提问作者JLC