Question

我想知道pandas.pivot_table是否可以同时接受两列并分别处理它们而不是分层处理。

说我有以下数据框：

 id    date   day  val
101   11/1/1   1   2.1
101   11/1/2   2   2.2
101   11/1/3   3   2.3
102   11/1/2   1   3.1
102   11/1/3   2   3.2
102   11/1/4   3   3.3

我希望结果如下：

      date                            day
 id  11/1/1  11/1/2  11/1/3  11/1/4    1   2   3
101   2.1     2.2      2.3     NaN    2.1 2.2 2.3
102   NaN     3.1      3.2     3.3    3.1 3.2 3.3

当我执行df.pivot_table(index='id', columns=['date','day'], values='val')时，它会将date和day集成到一个不是我想要的层次结构中。当然，我可以分别用date和day做两次并连接结果，但有没有更方便的方法一次这样做？

Answer 1

您可以进行2次pivot来电，concat来表示结果。

i = df.pivot('id', 'date', 'val')
j = df.pivot('id', 'day', 'val')

pd.concat([i, j], 1, keys=['date', 'day'])

      date                       day          
    11/1/1 11/1/2 11/1/3 11/1/4    1    2    3
id                                            
101    2.1    2.2    2.3    NaN  2.1  2.2  2.3
102    NaN    3.1    3.2    3.3  3.1  3.2  3.3

作为单个班轮 -

c = ['date', 'day']  # add more cols as needed
pd.concat([df.pivot('id', x, 'val') for x in c], axis=1, keys=c)

      date                       day          
    11/1/1 11/1/2 11/1/3 11/1/4    1    2    3
id                                            
101    2.1    2.2    2.3    NaN  2.1  2.2  2.3
102    NaN    3.1    3.2    3.3  3.1  3.2  3.3

Answer 2

一行....但为什么你需要它们在一行......

df.set_index(['id','val']).stack().to_frame('col').\
         reset_index(level='val').set_index('col',append=True).\
              unstack([-2,-1]).sort_index(1,level=1)
Out[69]: 
       val                                    
      date                       day          
col 11/1/1 11/1/2 11/1/3 11/1/4    1    2    3
id                                            
101    2.1    2.2    2.3    NaN  2.1  2.2  2.3
102    NaN    3.1    3.2    3.3  3.1  3.2  3.3

Pandas可以同时使用多列来旋转表格

2 个答案: