[英]Converting multiple rows to a single row based on key variable using pandas
输入数据集
Var1 Var2 Var3 Var4
101 XXX yyyy 12/10/2014
101 XYZ YTRT 13/10/2014
102 TTY UUUU 9/9/2014
102 YTY IUYY 10/10/2014
预期的输出数据集:
Var1 Var2 Var3 Var4
101 XXX,XYZ yyyy,YTRI 12/10/2014, 13/10/2014
102 TTY,YTY UUUU,IUYY 9/9/2014, 10/10/2014
如何通过熊猫编程获得预期的数据集?
一种方法是:
import pandas as pd
data = {'Var1': {0: 101, 1: 101, 2: 102, 3: 102},
'Var2': {0: 'XXX', 1: 'XYZ', 2: 'TTY', 3: 'YTY'},
'Var3': {0: 'yyyy', 1: 'YTRT', 2: 'UUUU', 3: 'IUYY'},
'Var4': {0: '12/10/2014', 1: '13/10/2014', 2: '9/9/2014', 3: '10/10/2014'}}
df = pd.DataFrame(data)
df.set_index('Var1', inplace=True)
print df
Var2 Var3 Var4
Var1
101 XXX yyyy 12/10/2014
101 XYZ YTRT 13/10/2014
102 TTY UUUU 9/9/2014
102 YTY IUYY 10/10/2014
f = lambda x: ','.join(x)
print df.groupby(level='Var1', as_index=True).transform(f).drop_duplicates().reset_index()
Var1 Var2 Var3 Var4
0 101 XXX,XYZ yyyy,YTRT 12/10/2014,13/10/2014
1 102 TTY,YTY UUUU,IUYY 9/9/2014,10/10/2014
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.