Python Dataframe: Assign values to a column based on condition?

Question

I have a df with name and status with present and absent as values.

            name     status
    0      anthony   present
    1      anthony   absent
    2      mason     present
    3      donny     present
    4      donny     absent
    5      donny     absent
    6      paul      present
    7      paul      present

I'm trying to put status - True or False for each name.

If one of the name has absent their status will be False

Expected Output:

            name     output
    0      anthony   False
    1      mason     True
    2      donny     False
    3      paul      True

Quang Hoang · Accepted Answer · 2020-10-28 19:43:24Z

1

You can compare status to 'present' then get the mininum by name:

# or `all()` instead of `min()`
df['status'].eq('present').groupby(df['name']).min()

Output:

name
anthony    False
donny      False
mason       True
paul        True
Name: status, dtype: bool

answered Oct 28, 2020 at 19:43

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

Comments

wwnde · Accepted Answer · 2020-10-28 21:07:10Z

0

Another way is to groupby name and find if a name has present as the only status.

df.groupby('name')['status'].apply(lambda x: (x=='present').all())



    name
    anthony    False
    donny      False
    mason       True
    paul        True

answered Oct 28, 2020 at 21:07

wwnde

26.7k6 gold badges22 silver badges38 bronze badges

Collectives™ on Stack Overflow

Python Dataframe: Assign values to a column based on condition?

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related