Python/Pandas - Checking the list of values in pandas column for a condition

Question

I have a python program which process multiple files. Each file have customer id column based on value in city column. Some file has 8 digit customer id and some have 9 digit customer id. I need to do this -

if column city value is in ['Chicago', 'Newyork', 'Milwaukee', 'Dallas']:
    input_file['customer_id'] = input_file['customer_id'].str[0:2] + '-' + input_file['customer_id'].str[2:8]
Else:
    input_file['customer_id'] = 'K' + '-' + input_file['customer_id'].str[0:3] + '-' + input_file['customer_id'].str[3:8]

Input

id cusotmerid  city
1  89898988    Dallas
2  898989889   Houston

Output

id cusotmerid     city
1  89-898988      Dallas
2  K-898-989889   Houston

How can I achieve it.

BENY · Accepted Answer · 2020-09-30 01:13:14Z

1

Check with np.where

l =  ['Chicago', 'Newyork', 'Milwaukee', 'Dallas']
s1 = df['customer_id'].str[0:2] + '-' + df['customer_id'].str[2:8]
s2 = 'K' + '-' + df['customer_id'].str[0:3] + '-' + df['customer_id'].str[3:8]
df['cusotmerid'] = np.where(df['city'].isin(l), s1, s2)

answered Sep 30, 2020 at 1:13

BENY

324k22 gold badges176 silver badges250 bronze badges

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

Collectives™ on Stack Overflow

Python/Pandas - Checking the list of values in pandas column for a condition

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related