python - 每组最大值

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']具有最大值的新对象。BA

输出应该是：

      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()

标签： pythonpandas