Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

adding a pandas column with multiple conditions

# create a list of  conditions
conditions = [
    (df['likes_count'] <= 2),
    (df['likes_count'] > 2) & (df['likes_count'] <= 9),
    (df['likes_count'] > 9) & (df['likes_count'] <= 15),
    (df['likes_count'] > 15)
    ]

# create a list of the values we want to assign for each condition
choices = ['tier_4', 'tier_3', 'tier_2', 'tier_1']

# create a new column and use np.select to assign values to it using our lists as arguments
df['tier'] = np.select(conditions, choices)

# display updated DataFrame
df.head()
Source by www.dataquest.io #
 
PREVIOUS NEXT
Tagged: #adding #pandas #column #multiple #conditions
ADD COMMENT
Topic
Name
4+6 =