Добавление сводного столбца MultiIndex в начале DataFrame - PullRequest
Новая
       10

Добавление сводного столбца MultiIndex в начале DataFrame

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

Если у меня есть Dataframe, и я бы хотел в начале вставить итоговый столбец, я могу запустить 

df.insert(0, 'Average', df.mean(axis='columns'))

Скажем, у меня есть мультииндексированный фрейм данных в форме 

df = pd.DataFrame()
for l1 in ('a', 'b'):
    for l2 in ('one', 'two'):
        df[l1, l2] = np.random.random(size=5)
df.columns = pd.MultiIndex.from_tuples(df.columns, names=['L1', 'L2'])

L1         a                   b          
L2       one       two       one       two
0   0.585409  0.563870  0.535770  0.868020
1   0.404546  0.102884  0.254945  0.362751
2   0.475362  0.601632  0.476761  0.665126 
3   0.926288  0.615655  0.257977  0.668778
4   0.509069  0.706685  0.355842  0.891862

Как мне добавить среднее значение всех столбцов one и всех столбцов two к первым двум столбцам этого DataFrame и назвать его 'Average'?

РЕДАКТИРОВАТЬ: Ожидаемый результат будет df.mean(level=1, axis=1) но вставляется в первые два столбца кадра с меткой L1 'Average'.Я надеялся, что будет работать следующее: 

df.insert(0, 'Average', df.mean(level=1, axis=1))

1 Ответ

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

IIUC, вам просто нужно groupby вычислить среднее значение, а затем немного поработать со столбцами полученного ряда: 

s = df.groupby(level=1, axis=1).mean()
s.columns = pd.MultiIndex.from_product([['Average'], s.columns])
pd.concat([s, df], 1)


    Average                   a                   b
        one       two       one       two       one       two
0  0.517939  0.713116  0.531990  0.578338  0.503889  0.847894
1  0.571197  0.676809  0.698986  0.425227  0.443409  0.928391
2  0.689653  0.399053  0.843179  0.069174  0.536126  0.728931
3  0.288367  0.197891  0.026974  0.026774  0.549761  0.369009
4  0.449904  0.590919  0.372560  0.556332  0.527247  0.625506
...