Pandas Переименование одной строки в MultiIndex с помощью Tuple - PullRequest
Новая
       69

Pandas Переименование одной строки в MultiIndex с помощью Tuple

2 голосов
/ 25 февраля 2020

Я пытаюсь переименовать одну строку pandas кадра данных по его кортежу.

Например: 

import pandas as pd
df = pd.DataFrame(data={'i1':[0,0,0,0,1,1,1,1],
                       'i2':[0,1,2,3,0,1,2,3],
                       'x':[1.,2.,3.,4.,5.,6.,7.,8.],
                       'y':[9,10,11,12,13,14,15,16]})
df.set_index(['i1','i2'], inplace=True)

Создает df: 

         x   y
i1 i2         
0  0   1.0   9
   1   2.0  10
   2   3.0  11
   3   4.0  12
1  0   5.0  13
   1   6.0  14
   2   7.0  15
   3   8.0  16

Я бы хотел использовать что-то вроде: df.rename(index={(0,1):(0,9)},inplace=True), чтобы получить: 

         x   y
i1 i2         
0  0   1.0   9
   9   2.0  10 <-- new key
   2   3.0  11
   3   4.0  12
1  0   5.0  13
   1   6.0  14
   2   7.0  15
   3   8.0  16

Команда выполняется без выдачи ошибки, но возвращает тот же самый df без изменений.

Это также возвращает тот же самый df: df.rename(index={pd.IndexSlice[0,1]:pd.IndexSlice[0,9]},inplace=True)

Это будет близко к желаемому эффект: 

df.loc[(0,9),:] = df.loc[(0,1),:]
df.drop(index=(0,1),inplace=True)

, но если порядок строк имеет значение, будет трудно привести его в правильное положение, и, возможно, будет довольно медленным, если df станет большим.

Я используя Pandas 1.0.1, python 3.7. Какие-либо предложения? Заранее спасибо.

1 Ответ

2 голосов
/ 25 февраля 2020

Возможное решение с пониманием списка и MultiIndex.from_tuples: 

L = [(0,9) if x == (0,1) else x for x in df.index]
df.index = pd.MultiIndex.from_tuples(L, names=df.index.names)
print (df)
         x   y
i1 i2         
0  0   1.0   9
   9   2.0  10
   2   3.0  11
   3   4.0  12
1  0   5.0  13
   1   6.0  14
   2   7.0  15
   3   8.0  16
...