pandas用apply创建条件列c时计算异常，无法匹配4类预设取值

• 运算符使用错误：你在单行标量判断的场景下，误用了pandas多行布尔筛选的位运算符&，且&的优先级高于比较运算符==，未加括号的情况下会先执行位运算，完全破坏了「a等于某值且b等于某值」的判断逻辑。
• 条件分支结构错误：连续写了4个独立的if而非if-elif链式判断，只要第一个逻辑错误的if命中就直接返回，后续分支永远不会执行，最终出现匹配错误、误返回0值的问题。

方案1：修正原有自定义函数 #

单行取标量判断直接用逻辑与and即可，无需额外加括号，同时调整分支为互斥的if-elif结构：

def test_func(data):
    if data['a'] == 0 and data['b'] == 0:
        return 1
    elif data['a'] == 0 and data['b'] == 1:
        return 2
    elif data['a'] == 1 and data['b'] == 0:
        return 3
    elif data['a'] == 1 and data['b'] == 1:
        return 4
    else:
        return 0

data['c'] = data.apply(test_func, axis=1)

如果要保留位运算符&，必须给每个比较条件单独加括号，写法如下：

def test_func(data):
    if (data['a'] == 0) & (data['b'] == 0):
        return 1
    elif (data['a'] == 0) & (data['b'] == 1):
        return 2
    elif (data['a'] == 1) & (data['b'] == 0):
        return 3
    elif (data['a'] == 1) & (data['b'] == 1):
        return 4
    else:
        return 0

方案2：高效向量化写法（无需apply） #

由于a、b都是0/1二元变量，可直接通过算术运算快速生成c列，性能远高于逐行apply：

# 对应映射关系：(a,b)=(0,0)→1、(0,1)→2、(1,0)→3、(1,1)→4
# 公式推导：a的权重为2，b的权重为1，加1刚好偏移到1-4的目标区间
data['c'] = 2 * data['a'] + data['b'] + 1

该写法无判断开销，适配所有二元组合，完全覆盖你定义的映射规则，不会出现异常返回值。

运行修正代码后，所有(a,b)组合都会正确匹配对应的c值，和你预期的输出完全一致，不会出现误返回0、匹配错误的问题。

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