Set Index Pandas

Set index в pandas используется для установки одного или нескольких столбцов как индекса датафрейма. Индекс позволяет быстро и удобно обращаться к определенным строкам или группам строк.

Пример 1:

Предположим, у нас есть датафрейм, содержащий информацию о студентах:

python
import pandas as pd
data = {'Имя': ['Алексей', 'Иван', 'Мария', 'Елена'],
        'Возраст': [20, 21, 19, 20],
        'Группа': [1, 2, 1, 2]}
df = pd.DataFrame(data)
print(df)

       Имя  Возраст  Группа
0  Алексей       20       1
1     Иван       21       2
2    Мария       19       1
3    Елена       20       2

Мы хотим установить столбец "Имя" в качестве индекса. Для этого используем метод `set_index()`:

python
df.set_index('Имя', inplace=True)
print(df)

         Возраст  Группа
Имя                      
Алексей       20       1
Иван          21       2
Мария         19       1
Елена         20       2

Теперь столбец "Имя" стал индексом датафрейма, что позволяет обращаться к строкам по имени.

Пример 2:

Иногда бывает необходимо установить несколько столбцов в качестве индекса. Рассмотрим следующий пример:

python
import pandas as pd
data = {'Имя': ['Алексей', 'Иван', 'Мария', 'Елена'],
        'Возраст': [20, 21, 19, 20],
        'Группа': [1, 2, 1, 2],
        'Рейтинг': [85, 90, 92, 88]}
df = pd.DataFrame(data)
print(df)

       Имя  Возраст  Группа  Рейтинг
0  Алексей       20       1       85
1     Иван       21       2       90
2    Мария       19       1       92
3    Елена       20       2       88

Хотим установить столбцы "Имя" и "Группа" как индекс. Для этого передаем список столбцов в метод `set_index()`:

python
df.set_index(['Имя', 'Группа'], inplace=True)
print(df)

                Возраст  Рейтинг
Имя     Группа                  
Алексей 1           20       85
Иван    2           21       90
Мария   1           19       92
Елена   2           20       88

Теперь датафрейм имеет двухуровневый индекс, состоящий из столбцов "Имя" и "Группа". Это позволяет обращаться к строкам по комбинации имени и группы.

Set index в pandas очень удобен и полезен при работе с датафреймами, так как он облегчает доступ и манипуляции с данными.