Retaining Group by character values in pandas dataframe

Question

import pandas as pd

df = pd.DataFrame({\'A\': [1, 1, 1, 2, 2, 2], \'B\': [\'sam\', \'anu\', \'rita\', \'first\', \'mid\', \'last\']}) print(df) enter image descriptio