Pandas按列反向列值

时间:2017-07-14 10:04:41

标签: python pandas sorting dataframe columnsorting

我想在我的数据框中反转列值,但仅限于个别" 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

熊猫大师,请指教我:)

1 个答案:

答案 0 :(得分:4)

您可以使用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