Вы можете использовать следующий синтаксис, чтобы установить столбец в pandas DataFrame в качестве индекса:
#set one column as index
df.set_index('col1')
#set multiple columns as multi index
df.set_index(['col1', 'col2'])
В следующих примерах показано, как использовать этот синтаксис на практике со следующим кадром данных:
import pandas as pd
#create DataFrame
df = pd.DataFrame({'points': [5, 7, 7, 9, 12, 9],
'assists': [11, 8, 10, 6, 6, 5],
'team': ['A', 'B', 'C', 'D', 'E', 'F'],
'conference': [1, 2, 3, 4, 5, 6]})
#view DataFrame
df
points assists team conference
0 5 11 A 1
1 7 8 B 2
2 7 10 C 3
3 9 6 D 4
4 12 6 E 5
5 9 5 F 6
Пример 1: установка одного столбца в качестве индекса
В следующем коде показано, как установить один столбец кадра данных pandas в качестве индекса:
df.set_index('team')
points assists conference
team
A 5 11 1
B 7 8 2
C 7 10 3
D 9 6 4
E 12 6 5
F 9 5 6
Пример 2. Установка нескольких столбцов в качестве индекса
В следующем коде показано, как установить несколько столбцов фрейма данных pandas в качестве мультииндекса :
df.set_index(['team', 'conference'])
points assists
team conference
A 1 5 11
B 2 7 8
C 3 7 10
D 4 9 6
E 5 12 6
F 6 9 5
Дополнительные ресурсы
Как переименовать индекс в Pandas DataFrame
Как удалить строки по индексу в Pandas
Как удалить столбцы по индексу в Pandas