python - Pandas 按组反转列值

Question

我想反转数据框中的列值，但仅限于单个“groupby”级别。您可以在下面找到一个最小的演示示例，我想在其中“翻转”属于同一字母 A、B 或 C 的值:

df = pd.DataFrame({"group":["A","A","A","B","B","B","B","C","C"],
                   "value": [1,3,2,4,4,2,3,2,5]})

  group  value
0     A      1
1     A      3
2     A      2
3     B      4
4     B      4
5     B      2
6     B      3
7     C      2
8     C      5

我想要的输出如下所示:(为了简洁起见，添加了列而不是替换了列)

  group  value  value_desired
0     A      1              2
1     A      3              3
2     A      2              1
3     B      4              3
4     B      4              2
5     B      2              4
6     B      3              4
7     C      2              5
8     C      5              2

一如既往，当我看不到合适的矢量风格方法时，我会为了最终输出而结束循环，但我当前的代码非常伤害我:

for i in list(set(df["group"].values.tolist())):
    reversed_group = df.loc[df["group"]==i,"value"].values.tolist()[::-1]
    df.loc[df["group"]==i,"value_desired"] = reversed_group

Pandas 大师，请指点一下:)

Answer 1

你可以使用transform

In [900]: df.groupby('group')['value'].transform(lambda x: x[::-1])
Out[900]:
0    2
1    3
2    1
3    3
4    2
5    4
6    4
7    5
8    2
Name: value, dtype: int64

详细信息

In [901]: df['value_desired'] = df.groupby('group')['value'].transform(lambda x: x[::-1])

In [902]: df
Out[902]:
  group  value  value_desired
0     A      1              2
1     A      3              3
2     A      2              1
3     B      4              3
4     B      4              2
5     B      2              4
6     B      3              4
7     C      2              5
8     C      5              2