Pandas create series using conditional

Question

I'm using the titanic dataset and have a created a series Famsize. I'd like to create a second series that outputs 'single' if famsize =1, 'small' if 1 < famsize < 5 and 'large' if famsize >=5.

   Famsize FamsizeDisc
     1         single
     2         small
     5         large

I've tried using np.where but as I have three outputs I haven't been able to find a solution.

Any suggestions?

do share what you've attempted so far.

parth
– parth

2017-10-05 11:00:50 +00:00
Commented Oct 5, 2017 at 11:00 — parth
– parth, Commented Oct 5, 2017 at 11:00

Bharath M Shetty · Accepted Answer · 2017-10-05 11:43:00Z

2

Its called binning so use pd.cut i.e

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

Output:

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

answered Oct 5, 2017 at 11:43

Bharath M Shetty

30.6k6 gold badges65 silver badges111 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

greg_data · Accepted Answer · 2017-10-05 11:51:28Z

1

Either you could create a function which does the mapping:

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)

Or you could use .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'

edited Oct 5, 2017 at 11:51

answered Oct 5, 2017 at 10:44

greg_data

2,29414 silver badges20 bronze badges

1 Comment

greg_data Over a year ago

My bad, hadn't reloaded the page to see your answer. I'll remove it from my answer and upvote yours, as it's clearly the more concise solution :D

Collectives™ on Stack Overflow

Pandas create series using conditional

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related