Groupby a pandas dataframe and create a new column by transform

Question

Imagine we have a DataFrame like df:

df = pd.DataFrame([(\'AL\', 2, 0.3),
                      (\'WY\', 4, 0.2),
                     (\'CO\', 1, 0.9),