每组最大值

Question

      A    B
0   Red  3.9
1   Red  4.1
2   Red  2.3
3  Blue -1.2
4  Blue -9.2
5  Blue -6.1

我想为每个df['C']组创建一个具有最大B值的新A。

输出应为：

      A    B    C
0   Red  3.9  4.1
1   Red  4.1  4.1
2   Red  2.3  4.1
3  Blue -1.2 -1.2
4  Blue -9.2 -1.2
5  Blue -6.1 -1.2

我尝试过：

df['C'] = df.groupby('A')['B'].max()