Mar*_*erc 3 python sorting dataframe pandas 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
Run Code Online (Sandbox Code Playgroud)
我想要的输出看起来像这样:(为了简洁起见,添加列而不是替换列)
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
Run Code Online (Sandbox Code Playgroud)
和往常一样,当我没有看到合适的向量样式方法时,我会为了最终输出而结束循环,但是我当前的代码非常伤害我:
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
Run Code Online (Sandbox Code Playgroud)
熊猫大师,请给我指路:)
您可以使用 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
Run Code Online (Sandbox Code Playgroud)
细节
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
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
1870 次 |
| 最近记录: |