joeyoung18 2021-03-25 13:13 采纳率: 100%
浏览 332
已采纳

DataFrame切片两列以上后,使用apply进行条件判断时报错

import numpy as np
import pandas as pd
df = pd.DataFrame(([1,2,3,4],[3,4,5,6]),columns = list('abcd'))
#希望做出的效果是:对a,b两列的内容进行判断,如果大于等于3则返回文本1,否则返回文本0
#在使用下方代码的时候会报错
df[['a','b']].apply(lambda x:'1' if x >=3 else '0')

#但只对单列进行同样操作的时候则没有问题
df['a'].apply(lambda x:'1' if x >=3 else '0')

请求各位大神解答一下疑惑...

如果没有办法这样切片两列以上再apply的话,那用什么替代方法比较优雅?

  • 写回答

1条回答 默认 最新

  • herosunly Python领域优质创作者 2021-03-25 13:39
    关注

    对整个DataFrame用apply,如果是多列处理,传参是axis=1。对于您这个问题,具体代码如下所示:

    import numpy as np
    import pandas as pd
    df = pd.DataFrame(([1,2,3,4],[3,4,5,6]),columns = list('abcd'))
    
    def process(row):
        if row['a'] >= 3 and row['b'] >= 3:
            return '1'
        else:
            return '0'
        
    df.apply(process, axis=1)
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 求差集那个函数有问题,有无佬可以解决
  • ¥15 【提问】基于Invest的水源涵养
  • ¥20 微信网友居然可以通过vx号找到我绑的手机号
  • ¥15 寻一个支付宝扫码远程授权登录的软件助手app
  • ¥15 解riccati方程组
  • ¥15 display:none;样式在嵌套结构中的已设置了display样式的元素上不起作用?
  • ¥15 使用rabbitMQ 消息队列作为url源进行多线程爬取时,总有几个url没有处理的问题。
  • ¥15 Ubuntu在安装序列比对软件STAR时出现报错如何解决
  • ¥50 树莓派安卓APK系统签名
  • ¥65 汇编语言除法溢出问题