Как узнать названия колонок в pandas? 🐼

    import pandas as pd
    
    # Создание DataFrame
    df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
    
    # Получение названий колонок
    column_names = df.columns
    
    # Вывод названий колонок
    print(column_names)
    

Как посмотреть название колонок pandas

В библиотеке pandas, чтобы посмотреть название колонок в датафрейме, можно использовать атрибут columns или метод columns.tolist(). Давайте рассмотрим примеры использования каждого из этих способов:

Использование атрибута "columns"

import pandas as pd

# Создаем простой датафрейм
data = {'Имя': ['Алиса', 'Боб', 'Клара'],
        'Возраст': [20, 25, 30],
        'Город': ['Москва', 'Санкт-Петербург', 'Екатеринбург']}

df = pd.DataFrame(data)

# Выводим название колонок
print(df.columns)
    

В результате выполнения данного кода мы получим:

Index(['Имя', 'Возраст', 'Город'], dtype='object')
    

Функция df.columns возвращает объект типа "Index", содержащий названия колонок датафрейма.

Использование метода "columns.tolist()"

import pandas as pd

# Создаем простой датафрейм
data = {'Имя': ['Алиса', 'Боб', 'Клара'],
        'Возраст': [20, 25, 30],
        'Город': ['Москва', 'Санкт-Петербург', 'Екатеринбург']}

df = pd.DataFrame(data)

# Выводим название колонок в виде списка
print(df.columns.tolist())
    

В результате выполнения данного кода мы получим:

['Имя', 'Возраст', 'Город']
    

Метод df.columns.tolist() преобразует объект типа "Index" в обычный список с названиями колонок.

Заключение

Теперь у вас есть два простых способа посмотреть название колонок в pandas. Вы можете использовать атрибут columns для получения объекта типа "Index" с названиями колонок или метод columns.tolist() для получения списка с названиями колонок. Оба этих способа достаточно просты в использовании и могут быть полезны при анализе данных с помощью pandas.