如何使用groupby函数根据类似的IDS创建行到列

Question

我有数据形式

arun 21-09-2017

raja 21-08-2016

arun 21-10-2017

raja 21-01-2017

我希望我的输入转换如下。

arun 21-09-2017 21-01-2017
raja 21-08-2016 21-01-2017

Answer 1

假设您正在使用pandas数据帧，则可以使用groupby获取所需的输出。

data = {'name':['arun','raja','arun','raja'],
       'date':['21-09-2017','21-08-2016','21-10-2017','21-01-2017']
       }

df1 = pd.DataFrame(data)

df1.groupby('name')['date'].apply(list).reset_index()

Out[1]:
   name                      date
0  arun  [21-09-2017, 21-10-2017]
1  raja  [21-08-2016, 21-01-2017]