python - 如何在python中使用if else语句对变量进行bin

Question

嗨，我的数据中有一个变量“年龄”，我想将它放入 python 中并将其保存到一个新变量中。像这样的东西”

data['new_age']= if data['age'] <= 16 then 'young'
                 if data['age'] > 16 & data['age'] <= 35 then 'adult'
                 else 'old' end

我怎么能用熊猫数据框做到这一点？

标签： pythonpandasif-statement

Answer 1

def new_age(r):
    if r['age']<=16:
        return 'young'
    if r['age'] <=35
        return 'adult'
    return 'old'

df['new_age'] = df.apply(lambda x: new_age(x),axis = 1)

完全不同意你的年龄划分，35岁是年轻人