how to set a new column based on group in pandas

Question

I have a dataframe as follows.

   ID  days group

0 1 5.3 A 1 2 6.0 A 2 3 6.7 A 3 4 5.5 B 4 5 6.2 B 5