如何根据日期拆分行？

Question

我有一个 table 这样的：

d = {
    "person_id" : [10,10,10,10,20,20],
    "type" : ["Exit", "Enter", "Exit", "Enter", "Exit", "Enter"],
    "exit_date": ["2022-01-01", "2022-01-02", "2022-01-05", "2022-01-07", "2022-02-01", "2022-02-05"]
}
pd.DataFrame(d)

我想将数据帧转换成这样：

d = {
    "person_id":[10, 10,20],
    "Exit": ["2022-01-01","2022-01-05", "2022-02-01"],
    "Enter": ["2022-01-02","2022-01-07", "2022-02-05"]
}
pd.DataFrame(d)

谢谢。

Answer 1

尝试：

>>> df.assign(index=df.groupby(['person_id', 'type']).cumcount()) \
      .pivot(['index', 'person_id'], 'type', 'exit_date') \
      .reset_index(level=1).to_dict('list')

{'person_id': [10, 20, 10],
 'Enter': ['2022-01-02', '2022-02-05', '2022-01-07'],
 'Exit': ['2022-01-01', '2022-02-01', '2022-01-05']}

Answer 2

这里只是替代. Using pivot_table，可以直接传一个石斑鱼：

(df.pivot_table(index=[df.groupby(['person_id', 'type']).cumcount(), 'person_id'],
                columns='type', values='exit_date', aggfunc='first')
   .droplevel(0).reset_index()
)

输出：

type  person_id       Enter        Exit
0            10  2022-01-02  2022-01-01
1            20  2022-02-05  2022-02-01
2            10  2022-01-07  2022-01-05

如何根据日期拆分行？

How to split rows with respect to date?

python

pandas

pivot-table