Pandas 按组反转列值

Question

我想反转数据框中的列值，但仅限于单个 "groupby" 级别。您可以在下面找到一个最小的演示示例，其中我想要 "flip" 属于相同字母 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

Pandas 按组反转列值

Pandas reverse column values groupwise

python

sorting

dataframe

pandas

columnsorting