熊猫使用条件创建系列

我正在使用泰坦尼克号数据集并创建了一系列Famsize。我想创建第二个系列，如果famsize = 1，则输出'单';如果输入'<'，则输出'小'，如果famsize> = 5，则输出'大'。熊猫使用条件创建系列

Famsize FamsizeDisc 
    1   single 
    2   small 
    5   large

我试过使用np.where但是因为我有三个输出我一直没有找到解决方案。

有什么建议吗？

来源

2017-10-05 Lonewoolf

确实分享了迄今为止尝试的内容。 – parth

其所谓binning所以使用pd.cut即

df['new'] = pd.cut(df['Famsize'],bins=[0,1,4,np.inf],labels=['single','small','large'])

输出：

 
    Famsize FamsizeDisc  new 
0  1  single single 
1  2  small small 
2  5  large large

来源

2017-10-05 11:43:00 Dark

要么你可以创建一个函数，它不映射：

def get_sizeDisc(x): 
    if x == 1: 
     return 'single' 
    elif x < 5: 
     return 'small' 
    elif x >= 5: 
     return 'large' 

df['FamsizeDisc'] = df.Famsize.apply(get_sizeDisc)

或者你可以使用.loc

df.loc[df.Famsize==1, 'FamsizeDisc'] = 'single' 
df.loc[df.Famsize.between(1,5, inclusive = False), 'FamsizeDisc'] = 'small' 
df.loc[df.Famsize>=5, 'FamsizeDisc'] = 'large'

来源

2017-10-05 10:44:45

这可以使用'pd.cut'在一行中完成。 – Dark

答案已更新。 –

但我补充说，作为一个答案:( – Dark

熊猫使用条件创建系列

回答

相关问题