如何在Python用户定义函数中使用列值?

2022-09-28 21:39:32 发布

您现在位置:Python中文网/ 问答频道 /正文

我想检查用户定义函数中某列的值,如果该列的值为1,它将返回1,如果该值为0,它将返回0。然后,我想用这些值创建一个新列

def count_developer(x): 
    if ['datascientist'] == "1":
        return 1 
    else:
        return 0

dat2['count_developer'] = dat2.apply(count_developer,axis=1)

但是,在新创建的列“count\u developer”中,我只得到“0”作为值,尽管列“datascientist”的值同时为0和1


Tags: 函数developerreturnif定义def检查用户countelseapplyaxisdat2datascientist
2条回答
网友
1楼 ·

试试这个:

def count_developer(x): 
    if ['datascientist'] == 1 or ['datascientist'] == "1":
        return 1 
    else:
        return 0
dat2['count_developer'] = dat2.apply(count_developer,axis=1)
网友
2楼 ·

尝试使用: 您正在返回0和1。因此它不会执行下一行

def count_developer(x): 
    if ['datascientist'] == "1":
        return_val = 1 
    else:
        return_val = 0
    dat2['count_developer'] = dat2.apply(count_developer,axis=1)
    return return_val

热门问题