Транспонировать столбцы в статистическую функцию в Pandas - PullRequest
0 голосов
/ 22 мая 2018

У меня есть агрегатная функция с group by для получения суммированных значений.Мой набор данных:

df=pd.DataFrame({"A":['a','a','a','a','a','a','b','b','b','b'],
         "Sales":[2,3,7,1,4,3,5,6,9,10],
         "Units":[12,2,2,33,6,2,4,8,3,5],
         "Week":[1,2,2,1,2,1,1,2,2,1]})

После этого я применяю функцию:

def my_agg(x):
    names = {
        'Sales': x['Sales'].sum(),
        'Units': x['Sales'].sum()
             } 

    return pd.Series(names, index=['Sales','Units'])
dfA= df.groupby(['A','Week']).apply(my_agg)

, которая дает мне вывод:

    Sales  Units
A Week              
a 1         6      6
  2        14     14
b 1        15     15
  2        15     15

Я хочу перенести неделюв столбцы.Вот так: ТРЕБУЕМЫЙ ВЫХОД:

   Week 1              2      
A      Sales  Units   Sales  Units        
a         6     6      14     14   
b        15    15      15     15

ТАКЖЕ, пожалуйста, предложите для ВЫХОДА 2:

           Sales         Units
A  Week   1            2
a         6    14      6      14   
b        15    15     15      15

Ответы [ 2 ]

0 голосов
/ 22 мая 2018

Выход 1

df.pivot_table(index='A', columns='Week', aggfunc='sum').swaplevel(1, 0, 1)

Week     1     2     1     2
     Sales Sales Units Units
A                           
a        6    14    47    10
b       15    15     9    11

Выход 2

df.pivot_table(index='A', columns='Week', aggfunc='sum')

     Sales     Units    
Week     1   2     1   2
A                       
a        6  14    47  10
b       15  15     9  11
0 голосов
/ 22 мая 2018

unstack с swaplevel

s=dfA.unstack()
s
Out[127]: 
     Sales     Units    
Week     1   2     1   2
A                       
a        6  14     6  14
b       15  15    15  15
s.swaplevel(0,1,axis=1).sort_index(level=0,axis=1)
Out[128]: 
Week     1           2      
     Sales Units Sales Units
A                           
a        6     6    14    14
b       15    15    15    15
...