用pandas apply 创造新列条件运行遇到奇怪的现象

发布于 2022-09-12 01:10:16 字数 892 浏览 14 评论 0

我的数据有两列,比如A列和B列吧。A列有PERMANENTLY WITHDRAWN和TEMPORARILY WITHDRAWN,以及其他值,B列有DRUG WITHDRAWN,DRUG INTERRUPTED和其他值。我想通过比较AB两列创建一个新列C,新列在一下条件下为"YES": 如果A列是PERMANENTLY WITHDRAWN而且B列是DRUG WITHDRAWN,或者A列是TEMPORARILY WITHDRAWN,B列是DRUG INTERRUPTED,或者A列和B列完全相等。我的代码如下:

def aeact_compare(x,y):
    if x.strip()  in ["PERMANENTLY WITHDRAWN\n","PERMANENTLY WITHDRAWN"] and y.strip() in ["DRUG WITHDRAWN\n","DRUG WITHDRAWN"]:
        return "YES"
    elif x.strip() in ["TEMPORARILY WITHDRAWN\n","TEMPORARILY WITHDRAWN"] and y.strip() in ["DRUG INTERRUPTED\n","DRUG INTERRUPTED"]:
        return "YES"
    elif x.strip() == y.strip():
        return "YES"
    else:
        return "NO"
sdb_pdb_trt["Action Taken Match?"] = sdb_pdb_trt.apply(lambda x : aerel_compare(x['Action Taken'],x['Lastest Action Taken with TRT']),axis=1)       
 

可是代码头两行的比较没起作用,请大家帮我看一下代码问题出在哪里?

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文