🐼 Как получить имена столбцов в pandas

Вы можете получить названия столбцов в pandas, используя атрибут columns. Он возвращает список со всеми именами столбцов в вашем DataFrame.

import pandas as pd

data = {'col1': [1, 2, 3], 'col2': ['a', 'b', 'c']}
df = pd.DataFrame(data)

column_names = df.columns.tolist()
print(column_names)

Как получить имена столбцов в pandas

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

Метод 1: Использование атрибута 'columns'

Первый способ - использование атрибута 'columns' у объекта DataFrame. Этот атрибут возвращает массив с именами всех столбцов в DataFrame. Давайте посмотрим на пример:

import pandas as pd

# Создание DataFrame
data = {'Name': ['John', 'Emily', 'Michael'],
        'Age': [25, 28, 32],
        'City': ['New York', 'London', 'Paris']}

df = pd.DataFrame(data)

# Получение имен столбцов с помощью атрибута 'columns'
column_names = df.columns

print(column_names)
    

В этом примере мы создаем DataFrame из словаря и выводим имена столбцов. Выходной результат будет следующим:

Index(['Name', 'Age', 'City'], dtype='object')
    

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

Второй способ - использование метода 'tolist()' в сочетании с атрибутом 'columns'. Этот метод преобразует массив имен столбцов в обычный список. Вот как это выглядит:

import pandas as pd

# Создание DataFrame
data = {'Name': ['John', 'Emily', 'Michael'],
        'Age': [25, 28, 32],
        'City': ['New York', 'London', 'Paris']}

df = pd.DataFrame(data)

# Получение имен столбцов с помощью метода 'columns.tolist()'
column_names = df.columns.tolist()

print(column_names)
    

В этом примере мы создаем DataFrame из словаря и выводим имена столбцов в виде списка. Результат будет таким же, как в предыдущем примере:

['Name', 'Age', 'City']
    

Метод 3: Использование индексации итерируемого объекта

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

import pandas as pd

# Создание DataFrame
data = {'Name': ['John', 'Emily', 'Michael'],
        'Age': [25, 28, 32],
        'City': ['New York', 'London', 'Paris']}

df = pd.DataFrame(data)

# Получение имен столбцов с помощью индексации итерируемого объекта
column_names = [column for column in df]

print(column_names)
    

В этом примере мы создаем DataFrame из словаря и используем индексацию итерируемого объекта для получения имен столбцов. Результат будет таким же, как и в предыдущих примерах:

['Name', 'Age', 'City']
    

Метод 4: Использование метода 'get_values()'

Четвертый способ - использование метода 'get_values()'. Этот метод возвращает все значения DataFrame в виде массива, включая имена столбцов. Вот пример:

import pandas as pd

# Создание DataFrame
data = {'Name': ['John', 'Emily', 'Michael'],
        'Age': [25, 28, 32],
        'City': ['New York', 'London', 'Paris']}

df = pd.DataFrame(data)

# Получение имен столбцов с помощью метода 'get_values()'
column_names = df.get_values()[0]

print(column_names)
    

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

['Name', 'Age', 'City']
    

Ключевые выводы

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

Теперь, когда вы знаете, как получить имена столбцов, вы можете использовать эти методы в своих проектах pandas и обрабатывать данные с уверенностью.