lmo*_*csi 2 python dataframe pandas
在 python 的 pandas 包中,我想进行分组,以便保持特定的顺序。下面的代码似乎可以做到这一点,但是有没有更快/更简单的方法?或者是否保证 pandas 的 group by 函数保持源数据框中记录的顺序?
import pandas as pd
sep = ' - '
df1 = pd.DataFrame([[1,"apple",1],[1,"banana",3],[1,"pear",2],[2,"lemon",3],[2,"orange",1],[2,"apple",2]])
df1.columns = ["id","product","seq"]
df1 = df1.sort_values(by=["id","seq"])
pd.DataFrame(df1.groupby('id',as_index=False)['product'].apply(lambda x: sep.join(x)))
Run Code Online (Sandbox Code Playgroud)
使用参数sort=False,也不DataFrame需要构造函数:
df2 = df1.groupby('id', sort=False)['product'].apply(sep.join).reset_index()
print (df2)
id product
0 1 apple - pear - banana
1 2 orange - apple - lemon
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
1881 次 |
| 最近记录: |