Pandas: как установить столбец в качестве индекса

Вы можете использовать следующий синтаксис, чтобы установить столбец в 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