Как вывести названия колонок в pandas: пошаговое руководство для начинающих

Чтобы вывести названия колонок в объекте DataFrame в pandas, можно использовать атрибут columns. Вот пример:

import pandas as pd

# Создаем DataFrame
data = {'Column1': [1, 2, 3],
        'Column2': ['a', 'b', 'c'],
        'Column3': [True, False, True]}

df = pd.DataFrame(data)

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

Результатом этого кода будет вывод названий колонок:

Index(['Column1', 'Column2', 'Column3'], dtype='object')

Вывести названия колонок в pandas

Pandas - это мощная библиотека для обработки и анализа данных в Python. Одной из первых задач, с которыми мы сталкиваемся при работе с данными, является вывод названий колонок в нашей таблице. Давайте рассмотрим несколько способов, которые помогут нам справиться с этой задачей.

Метод columns

Самый простой способ вывести названия колонок в pandas - использовать метод columns. Для этого нам необходимо обратиться к свойству columns нашего DataFrame. Ниже приведен пример кода:

import pandas as pd

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

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

Пример вывода:

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

Метод columns возвращает объект типа Index, содержащий названия колонок в нашем DataFrame.

Метод keys

Второй способ - использовать метод keys. Нам необходимо обратиться к свойству keys у объекта columns. Ниже приведен пример кода:

import pandas as pd

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

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

Пример вывода:

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

Метод keys также возвращает объект типа Index, содержащий названия колонок в нашем DataFrame.

Атрибут columns и tolist()

Третий способ - использовать атрибут columns и метод tolist(). Мы можем преобразовать объект типа Index в список, чтобы получить названия колонок. Ниже приведен пример кода:

import pandas as pd

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

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

Пример вывода:

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

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

Атрибут columns и список

Четвертый способ - использовать атрибут columns и преобразовать его в список. Пример кода:

import pandas as pd

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

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

Пример вывода:

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

Преобразование объекта типа Index в список выполняется при помощи функции list().

Атрибут columns и for-цикл

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

import pandas as pd

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

# Выводим названия колонок
for column in df.columns:
    print(column)

Пример вывода:

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

Цикл for позволяет нам итерироваться по названиям колонок и выводить их по одному.

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