Question

我有一个数据框，可以从下面给出的代码中生成

df = pd.DataFrame({'person_id' :[1,2,3],'date1': ['12/31/2007','11/25/2009','10/06/2005'],'date1derived':[0,0,0],'val1':[2,4,6],'date2': ['12/31/2017','11/25/2019','10/06/2015'],'date2derived':[0,0,0],'val2':[1,3,5],'date3':['12/31/2027','11/25/2029','10/06/2025'],'date3derived':[0,0,0],'val3':[7,9,11]})

数据框如下图所示

我想将每个人的行保留为单独的行，而不是如上面的屏幕快照中所示的列。此外，我希望删除date1派生，date2派生的列。

我确实尝试了以下方法，但是它们没有提供预期的输出结果

1) df.set_index(['person_id']).stack()/unstack
2) df.set_index(['person_id','date1','date2','date3']).stack()/unstack()
3) df.set_index('person_id').unstack()/stack

如何获得这样的输出？我有600多个列，因此我认为手动编写列名称不会对我有帮助。

Answer 1

这是一个wide_to_long问题：

pd.wide_to_long(df, stubnames=['date', 'val'], i='person_id', j='grp').sort_index(level=0)

                     date  val
person_id grp                 
1         1    12/31/2007    2
          2    12/31/2017    1
          3    12/31/2027    7
2         1    11/25/2009    4
          2    11/25/2019    3
          3    11/25/2029    9
3         1    10/06/2005    6
          2    10/06/2015    5
          3    10/06/2025   11

要匹配您的预期输出：

df = pd.wide_to_long(df, stubnames=['date', 'val'], i='person_id', j='grp').sort_index(level=0)
df = df.reset_index(level=1, drop=True).reset_index()

   person_id        date  val
0          1  12/31/2007    2
1          1  12/31/2017    1
2          1  12/31/2027    7
3          2  11/25/2009    4
4          2  11/25/2019    3
5          2  11/25/2029    9
6          3  10/06/2005    6
7          3  10/06/2015    5
8          3  10/06/2025   11

Answer 2

您可以不使用 wide_to_long（）而使用 append（）

df2 = pd.DataFrame()
for i in range(1, 4):
    new_df = df[['person_id', f'date{i}', f'val{i}']]
    new_df.columns = ['person_id', 'date', 'val']
    df2 = df2.append(new_df)
df2.sort_values('person_id').reset_index(drop=True)

输出：

   person_id        date  val
0          1  12/31/2007    2
1          1  12/31/2017    1
2          1  12/31/2027    7
3          2  11/25/2009    4
4          2  11/25/2019    3
5          2  11/25/2029    9
6          3  10/06/2005    6
7          3  10/06/2015    5
8          3  10/06/2025   11

如何像使用熊猫一样将多列转换为多行但保留主键

2 个答案: