Длинные / широкие данные в широкие / длинные - PullRequest
Новая
       97

Длинные / широкие данные в широкие / длинные

0 голосов
/ 27 сентября 2018

У меня есть фрейм данных, который выглядит следующим образом: 

import pandas as pd
d = {'decil': ['1. decil','1. decil','2. decil','2. decil','3. decil','3. decil'],
    'kommune': ['AA','BB','AA','BB','AA','BB'],'2010':[44,25,242,423,845,962],
    '2011':[64,26,239,620,862,862]}    
df = pd.DataFrame(data=d)

Печать 

decil      kommune  2010  2011
1. decil   AA       44    64
1. decil   BB       25    26
2. decil   AA      242   239
2. decil   BB      423   620
3. decil   AA      845   862
3. decil   BB      962   862

Мой желаемый вывод выглядит примерно так 

 kommune  year  1. decil  2. decil  3. decil
 AA       2010        44       242       845
 AA       2011        64       239       862
 BB       2010        25       423       962
 BB       2011        25       620       862

То естьЯ ищу способ изменить столбец 'decil' с длинного на широкий формат и в то же время изменить столбцы года с широкого на длинный формат.Я попытался pd.pivot_table, циклы и unstack без какой-либо удачи.Есть ли какой-нибудь умный способ обойти это?Заранее спасибо за помощь.

1 Ответ

0 голосов
/ 27 сентября 2018

Используйте set_index с stack и unstack: 

df = (df.set_index(['decil','kommune'])
        .stack()
        .unstack(0)
        .reset_index()
        .rename_axis(None, axis=1))

print (df)
  kommune level_1  1. decil  2. decil  3. decil
0      AA    2010        44       242       845
1      AA    2011        64       239       862
2      BB    2010        25       423       962
3      BB    2011        26       620       862
...