🔍 Как вывести название столбцов в pandas? Учимся работать с названиями столбцов в Pandas | Урок по Pandas

    import pandas as pd
    
    # Создание DataFrame
    data = {'Имя': ['Алексей', 'Виктория', 'Дмитрий'],
            'Возраст': [25, 28, 32],
            'Работа': ['Инженер', 'Программист', 'Дизайнер']}
    df = pd.DataFrame(data)
    
    # Вывод названий столбцов
    print(df.columns)
    

как вывести название столбцов в pandas

Для отображения названий столбцов в pandas можно использовать несколько различных подходов. В этой статье мы рассмотрим несколько методов, которые позволят вам выводить названия столбцов в DataFrame.

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

Первый способ - использовать атрибут columns, который предоставляет доступ к списку названий столбцов:

import pandas as pd

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

df = pd.DataFrame(data)

# Вывод названий столбцов
column_names = df.columns
print(column_names)

В результате выполнения этого кода вы увидите вывод следующего списка:

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

Этот список содержит названия всех столбцов в DataFrame.

2. Использование метода columns.values

Второй способ - использовать метод values атрибута columns:

import pandas as pd

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

df = pd.DataFrame(data)

# Вывод названий столбцов
column_names = df.columns.values
print(column_names)

Результат будет таким же, как и в предыдущем примере:

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

Метод values возвращает названия столбцов в виде массива значений.

3. Использование метода list

Третий способ - преобразовать атрибут columns в список, используя метод list:

import pandas as pd

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

df = pd.DataFrame(data)

# Вывод названий столбцов
column_names = list(df.columns)
print(column_names)

Результат будет тем же, что и в предыдущих примерах:

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

Этот метод полезен, если вам нужно использовать названия столбцов в последующих операциях, которые требуют передачи списка вместо объекта Index.

4. Использование метода get_column_names

Последний способ - создать пользовательскую функцию, которая возвращает названия столбцов:

import pandas as pd

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

df = pd.DataFrame(data)

# Функция для получения названий столбцов
def get_column_names(dataframe):
    return list(dataframe.columns)

# Вывод названий столбцов
column_names = get_column_names(df)
print(column_names)

Результат будет таким же, как и в предыдущих примерах:

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

Этот метод позволяет вам создавать более гибкий код и настраивать функцию для получения нужного формата вывода.

Теперь вы знаете несколько способов вывода названий столбцов в pandas. Вы можете выбрать тот, который лучше всего соответствует вашим потребностям и предпочтениям.