Python 创建的新列的值不正确_Python_Loops_Variables_Apply

Python 创建的新列的值不正确

python loops variables

Python 创建的新列的值不正确,python,loops,variables,apply,Python,Loops,Variables,Apply,我有两个列a和b，都是二进制变量 a b 1 1 0 1 1 1 0 0 0 0 ... 1 1 0 1 1 0 0 0 0 0 在检查以下一些条件后，我需要创建一个新变量c： def test_func(data): if data['a'] == 0 & data['b'] == 0: return 1; if data['a'] == 0 & data[

我有两个列a和b，都是二进制变量

在检查以下一些条件后，我需要创建一个新变量c：

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

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

但我只在新的列c中得到了值1和4，也得到了值0。事实上，这四种组合都有（没有得到0的组合）。但是没有把它们放在新的c列中。我可以知道怎么做吗

Desired output:
    a b c
    1 1 4
    0 1 2
    1 1 4
    0 0 1
    0 0 1
...
    1 1 4
    0 1 2
    1 0 3 
    0 0 1
    0 0 1


But what i got:
        a b c
        1 1 4
        0 1 1
        1 1 4
        0 0 1
        0 0 1
    ...
        1 1 4
        0 1 1
        1 0 0 
        0 0 1
        0 0 1

我相信您试图做的事情中的问题来自于在代码中使用

，而我相信您想要的是

和

。这两条在python中不同（更多信息/示例），但基本上是

和

检查这两条语句的计算结果是否为

True

，而

是位运算符

因此，请尝试将代码更改为：

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

另一方面，您应该尽量避免在一个函数中使用太多的return语句，因为这可能会让人非常困惑。例如，您可以将其替换为在每个

if

块中更改的

result

变量，也可以将单个

if

语句更改为也使用

elif

我对这个基本例子的建议是：

def test_func(data):
    result = 0

    if data['a'] == 0 and data['b'] == 0:
        result = 1
    elif data['a'] == 0 and data['b'] == 1:
        result = 2
    elif data['a'] == 1 and data['b'] == 0:
        result = 3
    elif data['a'] == 1 and data['b'] == 1:
        result = 4
    
    return result

data={"a":1, "b":0} #for example
def test_func(*args):
    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"]=test_func(data)
print(data["c"])