Преобразование мультииндексных индексов в столбцы

У меня есть мультииндексный фрейм данных, например. в качестве:

import pandas as pd
import numpy as np

dates = pd.date_range('20130101',periods=3,freq='5s')
dates = dates.append(dates)

locations = list('AAABBB')
gascode = ['no2','o3','so2']*2

tup = pd.MultiIndex.from_tuples( zip(locations,gascode,dates), names=['Location','gas','Date'] )

data = pd.DataFrame(data=range(6),index=tup,columns=['val1'])

>>> data

Location gas Date                  val1         
A        no2 2013-01-01 00:00:00     0
         o3  2013-01-01 00:00:05     1
         so2 2013-01-01 00:00:10     2
B        no2 2013-01-01 00:00:00     3
         o3  2013-01-01 00:00:05     4
         so2 2013-01-01 00:00:10     5

Сохранение данных только из местоположения «А»:

data = data.xs(key='A',level='Location')

Теперь я хочу создать новые столбцы в соответствии с индексом «газ», чтобы получить результат:

Date                   no2   o3   so2
2013-01-01 00:00:00     0    nan  nan
2013-01-01 00:00:05     nan  1    nan
2013-01-01 00:00:10     nan  nan  2

Я попытался изменить индекс «дата», чтобы поставить «газ» в столбцы, но это не удалось.

data = data.pivot(index=data.index.get_level_values(level='date'),
                  columns=situ.index.get_level_values(level='gas'))

Я не понимаю, как этого добиться; может кто посоветует альтернативу?


pandas python-2.7 pivot multi-index
person tnknepp    schedule 15.01.2014    source источник

Ответы (1)


arrow_upward
4
arrow_downward

Вы можете unstack получить результат:

In [11]: data.xs(key='A', level='Location').unstack(0)
Out[11]: 
                     val1         
gas                   no2  o3  so2
Date                              
2013-01-01 00:00:00     0 NaN  NaN
2013-01-01 00:00:05   NaN   1  NaN
2013-01-01 00:00:10   NaN NaN    2

[3 rows x 3 columns]
person Andy Hayden    schedule 15.01.2014