How to create new column using condition in python DataFrame?

Question

Just Convert R Code into equivalent Python code.

Item_Type - Old Column Name Item_Type_new - New Column Name

perishable = c("Breads", "Breakfast", "Dairy", "Fruits and Vegetables", "Meat", "Seafood")

non_perishable = c("Baking Goods", "Canned", "Frozen Foods", "Hard Drinks", "Health and Hygiene", "Household", "Soft Drinks")

# create a new feature 'Item_Type_new'
combi[,Item_Type_new := ifelse(Item_Type %in% perishable, "perishable", ifelse(Item_Type %in% non_perishable, "non_perishable", "not_sure"))]

YOLO · Accepted Answer · 2019-01-02 10:47:34Z

2

With a simple function, you can apply on pandas dataframe:

def func(x, l1, l2):
    """
    x = input value
    l1 = list of perishables
    l2 = list of non-perishables
    """    
    if x in l1:
        return 'perishable'
    elif x in l2:
        return 'non-perishable'
    else:
        return 'not_sure'


perishable = ["Breads", "Breakfast", "Dairy", "Fruits and Vegetables", "Meat", "Seafood"]
non_perishable = ["Baking Goods", "Canned", "Frozen Foods", "Hard Drinks", "Health and Hygiene", "Household", "Soft Drinks"]

combi['Item_Type_new'] = combi.apply(lambda x: func(x, perishable, non_perishable), axis=1)

answered Jan 2, 2019 at 10:47

YOLO

22k5 gold badges25 silver badges42 bronze badges

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

Comments

Vivek Kalyanarangan · Accepted Answer · 2019-01-02 10:55:14Z

0

Use np.select() -

perishable = ["Breads", "Breakfast", "Dairy", "Fruits and Vegetables", "Meat", "Seafood"]

non_perishable = ["Baking Goods", "Canned", "Frozen Foods", "Hard Drinks", "Health and Hygiene", "Household", "Soft Drinks"]

conditions = [
    (combi['Item_Type'].isin(perishable)),
    (combi['Item_Type'].isin(non_perishable))]

choices = ['perishable', 'non_perishable']

combi['Item_Type_new'] = np.select(conditions, choices, default='non_perishable')

answered Jan 2, 2019 at 10:55

Vivek Kalyanarangan

9,0911 gold badge27 silver badges42 bronze badges

Collectives™ on Stack Overflow

How to create new column using condition in python DataFrame?

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