add string based on conditional formating of a dataframe

Question

data = {"marks":[1,2,3,4,5,6,7,8,9,10,11,12], "month":['jan','feb','mar','apr','may','jun','jul','aug','sep','oct','nov','dec']}
df2 = pd.DataFrame(data)

Till now I tried below but not getting as mentioned above:

for i in df2['month']:
   if (i=='jan' or i=='feb' or i=='mar'):
        df2['q'] = '1Q'
   else:
        df2['q']='other'

jezrael · Accepted Answer · 2021-11-25 07:21:18Z

2

Use Series.dt.quarter with convert column to datetimes and add q:

df2['new'] = 'q' + pd.to_datetime(df2['month'], format='%b').dt.quarter.astype(str)

Or use Series.map by dictionary:

d = {'jan':'q1', 'feb':'q1','mar':'q1',
     'apr':'q2','may':'q2', 'jun':'q2',
     'jul':'q3','aug':'q3', 'sep':'q3',
     'oct':'q4','nov':'q4', 'dec':'q4'}
    
df2['new'] = df2['month'].map(d)

edited Nov 25, 2021 at 7:21

answered Nov 25, 2021 at 7:14

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

add string based on conditional formating of a dataframe

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