🐼 Как убрать мультииндекс: простые способы для работы с pandas

Чтобы убрать мультииндекс в pandas, вы можете использовать метод reset_index().

import pandas as pd

# Создаем DataFrame с мультииндексом
data = {'Города': ['Москва', 'Санкт-Петербург', 'Новосибирск'],
        'Язык': ['Русский', 'Русский', 'Русский'],
        'Население': [12678079, 5398064, 1625631]}

df = pd.DataFrame(data)
df.set_index(['Города', 'Язык'], inplace=True)

# Убираем мультииндекс
df.reset_index(inplace=True)

После применения метода reset_index() мультииндекс будет удален, и DataFrame будет иметь простой числовой индекс.

pandas: как убрать мультииндекс

Многие разработчики данных используют библиотеку pandas для манипулирования и анализа данных. В pandas есть мощный инструмент для организации данных с использованием мультииндексов. Однако, есть ситуации, когда мы хотим преобразовать мультииндексы в обычные индексы. В этой статье мы рассмотрим, как это сделать.

Удаление мультииндекса

Для удаления мультииндекса в pandas мы можем использовать метод reset_index(). Этот метод преобразует мультииндексированный DataFrame в DataFrame с обычным индексом. При этом все уровни индекса станут столбцами. Давайте рассмотрим пример:

import pandas as pd

# Создание мультииндексированного DataFrame
data = {'A': [1, 2, 3],
        'B': [4, 5, 6]}
index = pd.MultiIndex.from_tuples([('a', 'x'), ('a', 'y'), ('b', 'z')], names=['Index_1', 'Index_2'])
df = pd.DataFrame(data, index=index)

# Вывод исходного DataFrame
print("Исходный DataFrame:")
print(df)

# Удаление мультииндекса
df_reset = df.reset_index()

# Вывод DataFrame после удаления мультииндекса
print("DataFrame после удаления мультииндекса:")
print(df_reset)
  

В этом примере мы создали мультииндексированный DataFrame с помощью функции pd.MultiIndex.from_tuples(). Затем мы использовали метод reset_index(), чтобы удалить мультииндекс и преобразовать его в DataFrame с обычным индексом. Результат выводится на экран.

Представление мультииндекса в виде столбцов

Если мы хотим сохранить значения мультииндекса в отдельных столбцах, мы можем использовать метод reset_index() с параметром drop=True. В этом случае мультииндекс будет удален, а значения индекса будут сохранены в столбцы. Пример:

import pandas as pd

# Создание мультииндексированного DataFrame
data = {'A': [1, 2, 3],
        'B': [4, 5, 6]}
index = pd.MultiIndex.from_tuples([('a', 'x'), ('a', 'y'), ('b', 'z')], names=['Index_1', 'Index_2'])
df = pd.DataFrame(data, index=index)

# Вывод исходного DataFrame
print("Исходный DataFrame:")
print(df)

# Удаление мультииндекса и сохранение его в виде столбцов
df_reset = df.reset_index(drop=True)

# Вывод DataFrame со значениями мультииндекса в виде столбцов
print("DataFrame со значениями мультииндекса в виде столбцов:")
print(df_reset)
  

В этом примере мы использовали параметр drop=True в методе reset_index(). Это позволяет удалить мультииндекс и сохранить значения индекса в отдельных столбцах. Результаты выводятся на экран.

Изменение имени столбца при удалении мультииндекса

По умолчанию столбец, содержащий значения уровней мультииндекса, будет иметь имя 'level_0', 'level_1' и так далее. Если мы хотим изменить это имя, мы можем использовать аргумент col_level в методе reset_index(). Пример:

import pandas as pd

# Создание мультииндексированного DataFrame
data = {'A': [1, 2, 3],
        'B': [4, 5, 6]}
index = pd.MultiIndex.from_tuples([('a', 'x'), ('a', 'y'), ('b', 'z')], names=['Index_1', 'Index_2'])
df = pd.DataFrame(data, index=index)

# Вывод исходного DataFrame
print("Исходный DataFrame:")
print(df)

# Удаление мультииндекса и изменение имени столбца
df_reset = df.reset_index(col_level=1)

# Вывод DataFrame после удаления мультииндекса и изменения имени столбца
print("DataFrame после удаления мультииндекса и изменения имени столбца:")
print(df_reset)  
  

В этом примере мы использовали аргумент col_level=1 в методе reset_index() для изменения имени столбца, содержащего значения уровней мультииндекса. Результаты выводятся на экран.

Заключение

В этой статье мы рассмотрели, как убрать мультииндекс в pandas. Мы использовали метод reset_index() для преобразования мультииндексированного DataFrame в обычный индекс. Мы также рассмотрели, как сохранить значения мультииндекса в отдельных столбцах и как изменить имя столбца, содержащего значения уровней мультииндекса. Теперь вы знаете, как управлять мультииндексами в pandas и преобразовывать их по своему усмотрению.