pandas里的loc并列条件索引

最新推荐文章于 2025-06-04 12:39:35 发布

原创最新推荐文章于 2025-06-04 12:39:35 发布 · 1.3w 阅读

4 ·

CC 4.0 BY-SA版权

日常问题专栏收录该内容

30 篇文章

订阅专栏

本文介绍了一种用于处理家庭规模特征的方法，该方法通过整合乘客的兄弟姐妹、配偶、父母和子女的数量来创建一个新特征“Fam_Size”。这个新特征将家庭规模分为三类：单身、小家庭和大家庭。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

def Family_feature(df):
    df['Fam_Size'] =df['SibSp']+df['Parch']
    df['Fam_Size'].loc[df['Fam_Size'] == 0] = 1
    df['Fam_Size'].loc[(df['Fam_Size'] > 1) & (df['Fam_Size'] <= 3)] = 2
#     df['Fam_Size'].loc[df['Fam_Size'] == 2] = 2
#     df['Fam_Size'].loc[df['Fam_Size'] == 3] = 2
    df['Fam_Size'].loc[df['Fam_Size'] > 3] = 3
    return df

df['Fam_Size'].loc[(df['Fam_Size'] > 1) & (df['Fam_Size'] <= 3)] = 2 只要中间加一个 & 符号，然后把两边的condition括号起来就行。