Pandas groupby,求和并填充原始数据帧

import pandas as pd
df_1 = pd.DataFrame({'color': ['blue', 'blue', 'yellow', 'yellow'], 'count': [1,3,4,5]})

color   count   
blue    1   
blue    3   
yellow  4   
yellow  5   

我想按颜色列和总和列分组,然后用结果填充原始数据帧。所以最终结果应该是这样的:

df_2 = pd.DataFrame({'color': ['blue', 'blue', 'yellow', 'yellow'], 'count': [1,3,4,5], 
                     'total_per_color': [4,4,9,9]})


color   count   total_per_color
blue    1       4
blue    3       4
yellow  4       9
yellow  5       9

我可以使用groupby和sum,然后使用pandas进行合并,但我想知道是否有更快的方法?在SQL中,可以使用 partition dplyr 和 mutate