Bemind
Учебник Python
Учебник Python
  • Python
    • Python Lists
      • Списковые включения в Python (Полное руководство с примерами)
      • Исправление ValueError: Слишком Много Значений Для Распаковки в Python
      • Как добавить словарь в список в Python
      • Как добавить строку в список в Python
      • Разница между массивами и списками в Python
      • Python: Различия между списками и кортежами
      • Как проверить, пуст ли список в Python
      • Как Итерировать (Циклически Проходить) По Списку в Python
      • Python List sort(): Подробное руководство по сортировке списков
      • Python List Extend: Как добавить несколько элементов в список
      • Python: Найти Индекс Всех Вхождений Элемента в Списке
      • Конвертация списка словарей в Pandas DataFrame
      • Генерация случайных чисел в Python
      • Поиск Индекса в Списке Python: Найти Первое, Последнее или Все Вхождения
      • Добавить в начало списка в Python (Вставить в начало)
      • Найти дубликаты в списке Python
      • Python: Умножение Списков (6 Различных Способов)
      • Python списки: Полный обзор
      • Python: Выбор случайного элемента из списка
      • 4 Способа Очистить Список в Python
      • Объяснение ошибки IndexError в Python: индекс списка выходит за пределы допустимого диапазона
      • Python: Получение индекса максимального элемента в списке
      • Python: Объединение списков – Слияние списков (8 способов)
      • Python: Проверка наличия элемента в списке
      • Python: Проверка наличия элемента в списке
      • Удаление элемента из списка в Python (pop, remove, del, clear)
      • Как перевернуть список в Python (6 способов)
      • Python: Замена элемента в списке (6 различных способов)
      • Python: Удаление дубликатов из списка (7 способов)
      • Python: Преобразование словаря в список кортежей (4 простых способа)
      • Python: Перемешать Список (Случайное Распределение Элементов Списка в Python)
      • Python: Пересечение двух списков
      • Python: Вычитание двух списков (4 простых способа!)
      • Длина или Размер Списка в Python: 5 Способов Узнать Длину Списка
      • Python: Транспонирование списка списков (5 простых способов!)
      • Python: Разделение списка (Пополам, на части)
      • Python: Комбинации списка (Получить все комбинации списка)
      • Python: Выравнивание списка списков (4 способа)
      • Разница между списками в Python: Нахождение разницы между двумя списками Python
      • Python: Найти среднее значение списка или списка списков
      • Как добавлять элементы в списки в Python – 4 простых способа!
      • Списковые включения в Python (Полное руководство с примерами)
      • 6 способов преобразовать список Python в строку
    • Python Dictionaries
      • Понимание словаря Python (с примерами)
      • Исправляем ValueError: Слишком Много Значений Для Распаковки в Python
      • Как добавить словарь в список в Python
      • Преобразование JSON в словарь Python
      • Полное руководство по вложенным словарям в Python
      • Копирование словаря в Python: Полное руководство
      • Конвертация списка словарей в Pandas DataFrame
      • Поиск дубликатов в списке Python
      • Полный обзор словарей в Python
      • Python: Добавление пары Ключ:Значение в Словарь
      • Python: Сортировка словаря по значениям
      • Слияние Словарей в Python – Комбинирование Словарей (7 Способов)
      • Python: Удаление Дубликатов из Списка (7 Способов)
      • Python: Преобразование словаря в список кортежей (4 простых способа)
      • Python: Красивая Печать Словаря (Dictionary) – 4 Способа
      • Python: Проверка пуст ли словарь (5 способов!)
      • Copy of Python: Проверка пуст ли словарь (5 способов!)
      • Python: Проверьте, существует ли ключ (или значение) в словаре (5 простых способов)
      • Python: Проверьте, существует ли ключ (или значение) в словаре (5 простых способов)
      • Python: Получение Ключа Словаря с Максимальным Значением (4 Способа)
      • Python: Удаление ключа из словаря (4 разных способа)
      • Как красиво вывести JSON-файл в Python (6 методов)
    • Python Strings
      • Python Капитализация Строк: Руководство по Преобразованию слов в Заглавные
      • Python strip: Как обрезать строку в Python
      • Python Обратная Строка: Руководство по Реверсированию Строк
      • Как Удалить Префикс или Суффикс из Строки в Python
      • Преобразование строки в формат заголовка в Python с помощью str.title()
      • Как добавить строку в список в Python
      • Python String startswith: Проверка, начинается ли строка с подстроки
      • Python String endswith: Проверка того, заканчивается ли строка подстрокой
      • Как удалить первый или последний символ из строки в Python
      • Как исправить: SyntaxError в Python - EOL при сканировании строкового литерала
      • Python String Contains: Проверка Наличия Подстроки в Строке
      • Как проверить, пустая ли строка в Python
      • Python Новая Строка и Как Печатать Без Переноса Строки
      • Как Конкатенировать Строки в Python: Полное Руководство
      • Python: Подсчет слов в строке или файле
      • Как создать список алфавита в Python
      • Python: Конкатенация строки и целого числа (Int)
      • Python: Сортировка строки (4 различных способа)
      • Python zfill и rjust: Добавление нулей в строку в Python
      • Python: Целое в Двоичное (Преобразование целого числа в двоичную строку)
      • Python rfind: Нахождение индекса последней подстроки в строке
      • Python SHA256 хеширование алгоритм: объяснение
      • Python: Усечение числа с плавающей точкой (6 различных способов)
      • Выбор между методами Python isdigit(), isnumeric() и isdecimal()
      • Python: Удаление специальных символов из строки
      • Python Приведение Строки к Нижнему Регистру с помощью .lower(), .casefold(), и .islower()
      • Python программа для проверки, является ли строка палиндромом (6 методов)
      • Python: Найдите все перестановки строки (3 легких способа!)
      • Python: Удаление пунктуации из строки (3 разных способа!)
      • Python: Найти индекс (или все индексы) подстроки в строке
      • Python: Удаление символов новой строки из строки
      • Python: Удаление символа из строки (4 способа)
      • Python: Количество вхождений в строке (4 способа!)
    • Встроенные функции Python
      • abs()
      • ascii()
      • aiter()
      • all()
      • any()
      • anext()
      • bin()
      • bool()
      • breakpoint()
      • bytearray()
      • bytes()
      • callable()
      • chr()
      • classmethod()
      • compile()
      • complex()
      • delattr()
      • dict()
      • dir()
      • divmod()
      • enumerate()
      • eval()
      • exec()
      • filter()
      • float()
      • format()
      • frozenset()
      • getattr()
      • globals()
      • hasattr()
      • hash()
      • help()
      • hex()
      • id()
      • input()
      • int()
      • issubclass()
      • iter()
      • len()
      • list()
      • locals()
      • map()
      • max()
      • memoryview()
      • min()
      • next()
      • object()
      • oct()
      • open()
      • ord()
      • pow()
      • print()
      • property()
      • range()
      • repr()
      • reversed()
      • round()
      • set()
      • setattr()
      • isinstance()
      • slice()
      • zip()
      • type()
      • sorted()
      • staticmethod()
      • str()
      • sum()
      • super()
      • tuple()
      • vars()
      • import()
    • Cобеседования Python. Разбор реальных вопросов.
    • Встроенные методы в Python
  • Учебники по Pandas и Numpy
    • Numpy
      • Функция активации ReLU для глубокого обучения: полное руководство по выпрямленному линейному блоку
      • Как нормализовать массивы NumPy (минимальное-максимальное масштабирование, Z-оценка, L2)
      • NumPy where: Условная обработка элементов массива
      • NumPy linspace: создание равномерно расположенных массивов с помощью np.linspace
      • Как рассчитать векторное произведение в Python
      • Разделение NumPy: Разделение массива NumPy на части
      • NumPy: Лучшие способы применения функции к массиву
      • NumPy full: Создание массивов с заданным значением
      • NumPy clip(): Ограничьте значения массива минимальным и максимальным значениями
      • NumPy cumsum: Расчет кумулятивных сумм массивов NumPy
      • Изучаем функцию np.histogram в NumPy: создаем гистограмму
      • NumPy arange(): Полное руководство (с примерами)
      • Руководство по индексации и срезам массивов NumPy: Полное руководство
      • NumPy argmin(): Получение индекса минимального значения в массивах
      • Выравнивание массива с помощью NumPy flatten
      • Объединение массивов NumPy по различным осям с использованием функции stack
      • Удаление размерности из массивов NumPy с помощью NumPy Squeeze
      • Функция np.repeat() NumPy: Повторение массивов NumPy
      • Использование функции NumPy.exp() для вычисления экспоненты
      • Реализация функции сигмоида на Python
      • NumPy Pad: Использование np.pad() для дополнения массивов и матриц
      • np.argmax(): Как использовать NumPy Argmax
      • NumPy logspace: Понимание функции np.logspace()
      • Использование NumPy Tile для Расположения Массивов
      • NumPy Zeros: Создание массивов и матриц с нулями в NumPy
      • Использование числа Пи в Python (NumPy и Math)
      • Распределение Нормального (Гауссова) Распределения в Numpy (Случайное Нормальное в Numpy)
      • NumPy для Data Science на Python
      • Расчет скалярного произведения с использованием Numpy в Python
      • Расчет натурального логарифма на Python
    • Pandas
      • Python сводные таблицы – Полное руководство
      • Изучение API стиля Pandas
      • Объяснение группировки по нескольким столбцам в Pandas с примерами
      • Удаление индексной колонки DataFrame в Pandas: Руководство с примерами
      • Pandas Quantile: Расчет процентилей в DataFrame
      • Как рассчитать скользящее среднее (среднее арифметическое) в Pandas
      • Руководство по использованию метода fillna в Pandas для работы с отсутствующими данными в DataFrame
      • Pandas unique(): Получение уникальных значений в DataFrame
      • Распакуйте Ваши Данные с Помощью Функции Melt в Pandas
      • Pandas date_range: Как Создать Диапазон Дат в Pandas
      • Сброс индекса в Pandas: как сбросить индекс в Pandas
      • Pandas replace() – Замена значений в DataFrame Pandas
      • Перемещение столбца DataFrame Pandas на позицию (В начало и в конец)
      • Учебное пособие по Python Pandas: полное руководство
      • Pandas: Замена NaN на нули
      • Преобразование DataFrame Pandas в файл Pickle
      • Конвертация Pandas DataFrame в JSON
      • Преобразование DataFrame Pandas в Словарь
      • Преобразование Pandas DataFrame в Список
      • Чтение файлов Parquet в Pandas с помощью pd.read_parquet
      • Pandas dropna(): Удаление отсутствующих записей и столбцов в DataFrame
      • Как Добавить Новый Столбец в DataFrame Pandas
      • Подсчёт уникальных значений в Pandas
      • Отображение всех столбцов и строк в DataFrame Pandas
      • Pandas to_excel: Запись DataFrames в файлы Excel
      • Как использовать Pandas для чтения файлов Excel в Python
      • Преобразование списка словарей в Pandas DataFrame
      • Как добавить/вставить строку в DataFrame Pandas
      • Диаграмма рассеяния в Pandas: Как создать диаграмму рассеяния в Pandas
      • Pandas to_datetime: Преобразование строки Pandas в дату и время
      • Введение в Pandas для Data Science
      • Индексация, Выборка и Присваивание Данных в Pandas
      • Суммирование и Анализ Pandas DataFrame
      • Преобразование столбцов Pandas с помощью map и apply
      • Группировка данных в Pandas с использованием cut и qcut
      • Дата и время в Pandas и Python
      • Очистка и подготовка данных в Pandas и Python
      • Pandas GroupBy: группировка, суммирование и агрегация данных в Python
      • Pandas Дата и Время в Части Даты (месяц, год и т.д.)
      • Pandas: Получение номера строки из DataFrame
      • Вычисление Взвешенного Среднего в Pandas и Python
      • Как перемешать строки Pandas Dataframe в Python
      • Pandas: количество столбцов (подсчет столбцов в DataFrame)
      • Pandas Sum: сложение столбцов и строк DataFrame
      • Pandas Diff: Вычисление Разницы Между Строками Pandas
      • Нормализация столбца или датафрейма Pandas (с использованием Pandas или sklearn)
      • Функция Rank в Pandas: Ранжирование данных в Dataframe (Эквивалент SQL row_number)
      • Pandas Describe: Описательная статистика вашего Dataframe
      • Pandas Shift: Перемещение столбца DataFrame вверх или вниз
      • 7 Способов Выполнения Выборки Данных в Pandas
      • Экспорт DataFrame Pandas в CSV файл – Использование .to_csv()
      • Pandas: Итерация по строкам DataFrame в Pandas
      • Pandas: Преобразование значений столбца в строки
      • Дисперсия в Pandas: Вычисление дисперсии столбца в Pandas Dataframe
      • Pandas: Создание DataFrame из списков (5 способов!)
      • Pandas Rename Index: Как переименовать индекс DataFrame в Pandas
      • Pandas: Подсчёт уникальных значений в объекте GroupBy
      • Pandas: Добавить дни к колонке с датами
      • Среднее в Pandas: Как рассчитать среднее для одной или нескольких колонок
      • Pandas Column to List – Конвертируйте колонку Pandas в список
      • Транспонирование Dataframe в Pandas
      • Python: Разделение DataFrame Pandas
      • Как получить имена столбцов в DataFrame Pandas
      • Pandas: Количество строк в DataFrame (6 способов)
      • Создание пустого DataFrame Pandas и добавление данных
      • Как переименовать столбцы в Pandas DataFrame (с примерами)
      • Изменение порядка столбцов в Pandas: использование метода reindex и метода insert
      • Pandas get_dummies (One-Hot кодирование), объяснение
      • Относительные и Абсолютные Частоты в Python и Pandas
      • Финансовый год – Определение финансового года в Pandas
      • Как сортировать данные в DataFrame Pandas
  • Учебники Matplotlib и Seaborn
    • Seaborn
      • Регрессионные графики в Seaborn с использованием regplot и lmplot
      • Seaborn residplot – Построение остатков линейной регрессии
      • Seaborn jointplot() – Создание совместных графиков в Seaborn
      • Seaborn displot – Распределенческие графики в Python
      • Seaborn ecdfplot – Эмпирические функции накопленного распределения
      • Seaborn rugplot – Визуализация маргинальных распределений
      • Seaborn kdeplot – Создание графиков оценки плотности ядра
      • Seaborn histplot – Создание Гистограмм в Seaborn
      • Seaborn catplot – Визуализация категориальных данных в Python
      • Средняя тенденция для категориальных данных в Seaborn Pointplot
      • Seaborn stripplot: Jitter Plots для распределений категориальных данных
      • Seaborn Countplot – Подсчет категориальных данных в Python
      • Seaborn swarmplot: Bee Swarm Plots для распределения категориальных данных
      • Скрипичные графики Seaborn в Python: Полное руководство
      • Настройка расположения легенд Seaborn, меток, текста и т.д.
      • Тепловая карта Seaborn: Полное руководство
      • Создание многосекционных сеток в Seaborn с помощью FacetGrid
      • Удаление рамки в Seaborn: Как работать с рамкой
      • Заголовки и метки осей в Seaborn: добавление и настройка
      • Как установить Seaborn в Python (Исправление: no module named seaborn)
      • Seaborn relplot – Создание точечных и линейных графиков
      • Полное руководство по созданию точечных диаграмм (scatter plots) в Python с использованием Seaborn
    • Matplotlib
      • Режим Retina в Matplotlib: Улучшение Качества Графиков
      • Как построить функцию в Python с использованием Matplotlib
      • Как создать 3D-диаграммы рассеяния в Matplotlib
      • Как изменить размер шрифта в графике Matplotlib
      • Установка размера маркера в точечных диаграммах Matplotlib
      • Как изменить размер графика и фигуры в Matplotlib
      • Как добавить названия в Matplotlib: Заголовок, Подзаголовок, Названия Осей
      • Pandas Scatter Plot: Как создать диаграмму рассеяния в Pandas
      • Построение графиков в Python с помощью Matplotlib
      • Диаграммы рассеяния Matplotlib – Все, что вам нужно знать
      • Диаграммы с столбцами в Matplotlib – Узнайте все, что вам нужно знать
      • Линейные диаграммы Matplotlib – Всё, что вам нужно знать
      • Построение гистограммы в Python с Matplotlib и Pandas
  • Алгоритмы
    • Алгоритм поиска в ширину (BFS) в Python
    • Алгоритм поиска в глубину (DFS) на Python
  • AI создает хедж-фонд для анализа акций на Python
Powered by GitBook
On this page
  • Загрузка примера DataFrame
  • Как получить имена столбцов из DataFrame Pandas
  • Как получить список имён столбцов Pandas из CSV-файла
  • Как проверить существование столбца в DataFrame Pandas
  • Как посчитать количество столбцов в DataFrame Pandas
  • Как получить словарь имён столбцов и их типов данных
  • Как получить имена столбцов определённых типов данных в Pandas
  • Как получить имена столбцов Pandas по индексу
  • Как получить имена столбцов Pandas в многоуровневом (MultiIndex) DataFrame
  • Как получить имена столбцов Pandas, удовлетворяющих условию
  • Как получить имена столбцов Pandas с пропущенными значениями
  • Как получить имена столбцов Pandas, содержащих дублирующиеся значения
  • Как получить алфавитный список имён столбцов в Pandas
  • Как получить имена столбцов, начинающихся с определённой буквы в Pandas
  • Часто задаваемые вопросы
  • Заключение
  • Дополнительные ресурсы
  1. Учебники по Pandas и Numpy
  2. Pandas

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

PreviousPython: Разделение DataFrame PandasNextPandas: Количество строк в DataFrame (6 способов)

Last updated 21 days ago

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

В данном уроке рассматриваются все возможные сценарии, и даются подробные инструкции. Умение получать и просматривать список всех столбцов в DataFrame на основе Pandas позволяет лучше понимать структуру данных и эффективнее с ними работать.

Оглавление

Загрузка примера DataFrame

Чтобы следовать материалу урока, мы предоставляем образец DataFrame, созданный с помощью Pandas. Если у вас есть собственные данные, вы можете использовать их — разумеется, результаты будут отличаться. Чтобы начать, просто скопируйте и вставьте код из блока ниже:

# Загрузка примера Pandas DataFrame
import pandas as pd

df = pd.read_csv('https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv')

print(df.head())

# Возвращает:
#         Date  Name  NumSold   Total  Active
# 0  01-Jan-23   Nik       14  329.67    True
# 1  02-Jan-23  Evan       12  475.71   False
# 2  03-Jan-23  Kyra       16  569.64    True
# 3  04-Jan-23  Kate       13  528.23   False
# 4  05-Jan-23   NaN       19  974.65    True

В приведённом выше блоке кода мы загрузили DataFrame Pandas с помощью функции pd.read_csv(). DataFrame содержит пять столбцов с разнородными типами данных, а также некоторые пропущенные значения.

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

Pandas предоставляет несколько удобных способов получить имена столбцов. Самый простой способ — передать DataFrame в функцию list(), которая возвращает список всех имён столбцов.

# Получить все названия столбцов в виде списка

import pandas as pd

df = pd.read_csv('https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv')

print(list(df))

# Возвращает:
# ['Date', 'Name', 'NumSold', 'Total', 'Active']

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

Кроме того, вы можете получить доступ ко всем именам столбцов в DataFrame Pandas с помощью атрибута .columns. Этот атрибут возвращает объект Pandas Index, содержащий все имена столбцов.

Хотя этот объект поддерживает итерацию, если вы хотите преобразовать его в список, необходимо либо использовать метод .tolist(), либо передать его в функцию list().

# Использование атрибута .columns для получения названий столбцов

import pandas as pd

df = pd.read_csv('https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv')

print(df.columns.tolist())

# Возвращает:
# ['Date', 'Name', 'NumSold', 'Total', 'Active']

Этот подход требует написания большего количества кода.

Почему стоит использовать атрибут .columns DataFrame Pandas для получения имён столбцов?

Атрибут .columns позволяет проверять принадлежность к столбцам DataFrame. Это упрощает проверку наличия определённого столбца, без необходимости создания отдельного списка имён столбцов.

Как получить список имён столбцов Pandas из CSV-файла

Pandas также позволяет очень легко получить список имён столбцов из CSV-файла. Для этого можно указать чтение только одной строки данных. После этого вы можете воспользоваться описанным выше методом для получения имён столбцов.

# Получить список названий столбцов Pandas из CSV

import pandas as pd

df = pd.read_csv(
    'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv',
    nrows=1
)

print(list(df))

# Возвращает:
# ['Date', 'Name', 'NumSold', 'Total', 'Active']

В приведённом выше примере мы загружаем только одну строку данных, используя аргумент nrows=1. Это позволяет загрузить минимально возможное количество данных (сэкономив память и время), при этом сохранив доступ к именам столбцов.

Как проверить существование столбца в DataFrame Pandas

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

Чтобы проверить, существует ли столбец в DataFrame, можно использовать ключевое слово in.

Давайте посмотрим, как можно проверить наличие столбца NumSold:

# Проверка существования столбца в Pandas DataFrame

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

print('NumSold' in df.columns)

# Возвращает: True

В приведённом выше коде мы использовали ключевое слово in, чтобы проверить, содержится ли строка 'NumSold' в итерируемом объекте df.columns.

Если мы хотим вывести сообщение в зависимости от того, существует столбец или нет, мы можем обернуть это в блок if-else, как показано ниже:

# Проверка существования столбца

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

if 'NumSold' in df.columns:
    print('Column exists!')
else:
    print("Column doesn't exist!")

# Возвращает: Column exists!

В приведённом выше примере мы написали блок if-else, который позволяет сообщить пользователю, существует ли указанный столбец.

Как посчитать количество столбцов в DataFrame Pandas

В этом разделе вы узнаете, как подсчитать количество столбцов в DataFrame Pandas. Объект, возвращаемый атрибутом df.columns, можно передать в функцию len(), чтобы определить количество столбцов.

Давайте посмотрим, как можно посчитать количество столбцов в DataFrame Pandas:

# Подсчет количества столбцов в Pandas DataFrame

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

print(len(df.columns))

# Возвращает: 5

Разберём, что мы сделали в приведённом выше блоке кода:

  1. Мы передали атрибут df.columns в функцию len().

  2. Это позволило получить длину объекта, похожего на список, который содержит имена столбцов.

В следующем разделе вы узнаете, как получить словарь имён столбцов и их соответствующих типов данных.

Как получить словарь имён столбцов и их типов данных

В этом разделе вы научитесь создавать словарь, в котором ключами являются имена столбцов, а значениями — их типы данных. Это помогает лучше понять структуру данных в вашем DataFrame.

Атрибут .dtypes библиотеки Pandas возвращает объект типа Series, содержащий имена столбцов и их типы данных. Давайте посмотрим, как это выглядит при выводе на экран:

# Получение типов данных Pandas DataFrame

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

print(df.dtypes)

# Возвращает:
# Date        object
# Name        object
# NumSold      int64
# Total      float64
# Active        bool
# dtype: object

Если вы хотите преобразовать этот объект Series в словарь Python, можно просто передать его в функцию dict(). Давайте посмотрим, как это выглядит:

# Получить словарь с названиями столбцов и типами данных

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

print(dict(df.dtypes))

# Возвращает:
# {'Date': dtype('O'), 'Name': dtype('O'), 'NumSold': dtype('int64'), 'Total': dtype('float64'), 'Active': dtype('bool')}

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

Как получить имена столбцов определённых типов данных в Pandas

Pandas позволяет очень легко получить список имён столбцов, относящихся к определённым типам данных. Это можно сделать с помощью метода .select_dtypes() и функции list(). Метод .select_dtypes() применяется к DataFrame и позволяет выбрать либо один, либо несколько типов данных.

Вы можете указать, какие типы данных включить, а какие — исключить. Давайте посмотрим, как выбрать только те столбцы, которые имеют булев тип (boolean):

# Получить названия столбцов с булевым типом данных

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

print(list(df.select_dtypes('bool')))

# Возвращает: ['Active']

В приведённом выше примере мы сузили DataFrame, оставив только столбцы с булевым типом данных. Затем мы получили список имён этих столбцов, передав результат в функцию list().

Функция .select_dtypes() из Pandas также позволяет легко выбрать все числовые столбцы. Вместо того чтобы передавать список конкретных числовых типов данных, можно просто указать 'number':

# Получить названия столбцов с числовым типом данных в Pandas

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

print(list(df.select_dtypes('number')))

# Возвращает: ['NumSold', 'Total']

Этот метод делает получение имён столбцов определённого типа (или нескольких типов) чрезвычайно простым.

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

В этом разделе вы узнаете, как получить имя столбца Pandas по его индексу (или индексам). Это позволяет узнать название столбца, находящегося на определённой позиции (или позициях).

Ранее вы научились получать список всех имён столбцов, передав DataFrame в функцию list(). Поскольку список поддерживает индексацию, мы можем обратиться к имени столбца по конкретному индексу.

Давайте посмотрим, как получить имя столбца, находящегося на второй позиции:

# Получить название столбца Pandas по индексу

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

print(list(df)[1])

# Возвращает: Name

В приведённом выше примере мы обратились к списку по индексу 1, чтобы получить второй элемент (поскольку индексация начинается с нуля).

Аналогично, мы можем использовать срезы для получения нескольких имён. Например, если нам нужно получить последние два имени столбцов, мы можем воспользоваться отрицательной индексацией:

# Получить названия последних 2 столбцов в Pandas DataFrame

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'

df = pd.read_csv(url)

print(list(df)[-2:])

# Возвращает: ['Total', 'Active']

Приведённый выше пример демонстрирует, что мы можем получить имена последних двух столбцов DataFrame в Pandas с помощью срезов списка.

Как получить имена столбцов Pandas в многоуровневом (MultiIndex) DataFrame

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

# Загрузка примера Pandas DataFrame с MultiIndex

import pandas as pd
import numpy as np

col = pd.MultiIndex.from_arrays([['one', 'one', 'two', 'two'],
                                ['a', 'b', 'a', 'b']])

df = pd.DataFrame(np.random.randn(4, 4), columns=col)
print(df)

# Возвращает:
#         one                 two
#           a         b         a         b
# 0  0.542193 -1.820652 -0.169705 -1.654994
# 1 -0.317596  0.446695  0.720554  0.801922
# 2  1.011754  0.126223 -0.340901  0.007976
# 3  0.657133 -1.029230 -1.262736  0.771902

Давайте посмотрим, что происходит, когда мы пытаемся получить имена столбцов с помощью ранее рассмотренных методов:

  1. Используя list(df)

  2. Используя df.columns

# Получение доступа к столбцам DataFrame с MultiIndex

import pandas as pd
import numpy as np

col = pd.MultiIndex.from_arrays([['one', 'one', 'two', 'two'],
                                ['a', 'b', 'a', 'b']])
df = pd.DataFrame(np.random.randn(4, 4), columns=col)

option1 = list(df)
option2 = df.columns

print(f'{option1=}')
print(f'{option2=}')

# Возвращает:
# option1=[('one', 'a'), ('one', 'b'), ('two', 'a'), ('two', 'b')]
# option2=MultiIndex([('one', 'a'),
#             ('one', 'b'),
#             ('two', 'a'),
#             ('two', 'b')],
#            )

Мы можем видеть, что оба варианта возвращают список, похожий на структуру, содержащую кортежи пар имён столбцов.

Pandas также позволяет очень просто получить доступ только к одному уровню имён столбцов. Например, если мы хотим получить доступ только к верхнему уровню столбцов, мы можем использовать метод .get_level_values() для атрибута .columns, как показано ниже:

# Получение доступа к одному уровню столбцов DataFrame с MultiIndex

import pandas as pd
import numpy as np

col = pd.MultiIndex.from_arrays([['one', 'one', 'two', 'two'],
                                ['a', 'b', 'a', 'b']])
df = pd.DataFrame(np.random.randn(4, 4), columns=col)

print(df.columns.get_level_values(0))

# Возвращает:
# Index(['one', 'one', 'two', 'two'], dtype='object')

В следующем разделе вы узнаете, как получить имена столбцов Pandas, удовлетворяющих определённому условию.

Как получить имена столбцов Pandas, удовлетворяющих условию

Чтобы получить список имён столбцов, для которых выполняется определённое условие, можно просто отфильтровать список столбцов с помощью булевой маски. Например, предположим, что мы хотим получить имена столбцов, в которых содержится буква 'a' (в любом регистре). Для этого можно написать следующий код:

# Получить столбцы Pandas, соответствующие условию

import pandas as pd
import re

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'
df = pd.read_csv(url)

cols = [col for col in list(df) if re.search(r'[aA]', col)]
print(cols)

# Возвращает: ['Date', 'Name', 'Total', 'Active']

Разберём, что мы сделали выше:

  1. Мы импортировали библиотеку Pandas и модуль для работы с регулярными выражениями re.

  2. С помощью генератора списка (list comprehension) мы отфильтровали только те столбцы, в названиях которых содержится буква 'a' (в любом регистре).

  3. Вывели полученный список на экран.

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

Как получить имена столбцов Pandas с пропущенными значениями

Чтобы получить список столбцов, содержащих пропущенные значения, можно просто выполнить срез объекта df.columns, добавив булево условие. Для определения наличия пропусков в данных используется комбинация методов .isna().any().

Давайте посмотрим, как это выглядит:

# Получить названия столбцов Pandas со значениями NaN

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'
df = pd.read_csv(url)

print(df.columns[df.isna().any()])

# Возвращает:
# Index(['Name'], dtype='object')

Из приведённого выше блока кода видно, что мы смогли отфильтровать список столбцов с помощью булевой маски. Массив, возвращаемый методом df.isna().any(), содержит только булевые значения. Это означает, что любой столбец, содержащий хотя бы одно пропущенное значение, помечается как True, а столбцы без пропусков — как False.

Как получить имена столбцов Pandas, содержащих дублирующиеся значения

Получение имён столбцов, содержащих дублирующиеся значения, выполняется похожим образом. Мы также применяем маску к массиву имён столбцов, возвращаемому df.columns. Эта маска представляет собой булев массив, указывающий, в каких столбцах есть дубликаты.

Давайте посмотрим, как это выглядит:

# Получить названия столбцов Pandas с повторяющимися значениями

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'
df = pd.read_csv(url)
df.loc[1, 'Name'] = 'Nik'

mask = df.apply(lambda x: x.duplicated().any(), axis=0)
print(df.columns[mask])

# Возвращает: Index(['Name', 'Active'], dtype='object')

Приведённый выше блок кода выглядит немного сложнее, поэтому давайте разберём, что мы сделали:

  1. Мы добавили новое дублирующееся значение в DataFrame с помощью акцессора .loc.

  2. Мы создали переменную mask, которая применяет функцию к каждому столбцу для определения, содержат ли они дублирующиеся значения.

  3. Затем мы применили эту маску к атрибуту df.columns, чтобы отфильтровать список столбцов.

В следующем разделе вы узнаете, как получить алфавитный список имён столбцов в Pandas.

Как получить алфавитный список имён столбцов в Pandas

Pandas позволяет очень легко получить отсортированный по алфавиту список имён столбцов с помощью встроенной функции sorted(). Поскольку эта функция возвращает отсортированный список, мы можем просто передать в неё все имена столбцов.

Давайте посмотрим, как получить алфавитный список имён столбцов в Pandas:

# Получить отсортированный по алфавиту список названий столбцов в Pandas

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'
df = pd.read_csv(url)

print(sorted(df))

# Возвращает:
# ['Active', 'Date', 'Name', 'NumSold', 'Total']

Преимущество этого подхода в том, что он требует совсем немного кода. Аналогично, вы могли бы отсортировать список, полученный через list(df) или df.columns, но использование функции sorted() позволяет сэкономить время и писать меньше кода!

Как получить имена столбцов, начинающихся с определённой буквы в Pandas

В этом разделе вы узнаете, как получить имена столбцов DataFrame Pandas, которые начинаются с определённой буквы. Для этого мы можем применить булеву маску к массиву, возвращаемому атрибутом df.columns.

Для создания такой маски можно использовать метод .str.startswith(), который применяется к массиву строк. Этот метод возвращает True для тех значений, которые начинаются с указанной буквы.

Давайте посмотрим, как получить имена столбцов, начинающихся с буквы 'N':

# Получить названия столбцов, начинающихся с определенной буквы, в Pandas

import pandas as pd

url = 'https://raw.githubusercontent.com/datagy/mediumdata/master/PandasColumns.csv'
df = pd.read_csv(url)

print(df.columns[df.columns.str.startswith('N')])

# Возвращает:
# Index(['Name', 'NumSold'], dtype='object')

Часто задаваемые вопросы

Как получить список всех имён столбцов в DataFrame Pandas? Самый простой способ — использовать list(df). Также можно воспользоваться атрибутом df.columns.


Заключение

В этом подробном руководстве вы узнали о множестве способов получения имён столбцов в DataFrame Pandas. Сначала вы познакомились с основными методами извлечения всех имён столбцов, а затем научились применять более специфические подходы. Например, вы узнали, как получить имена столбцов определённого типа данных, содержащих пропущенные значения, дубликаты и многое другое.


Дополнительные ресурсы

Чтобы узнать больше по смежным темам, ознакомьтесь со следующими уроками:

  • Как использовать Pandas для чтения Excel-файлов в Python

  • Pandas to_excel: Сохранение DataFrame в Excel-файл

  • Pandas GroupBy: Группировка, агрегация и обобщение данных в Python

  • Работа с датой и временем в Pandas и Python

Официальная документация к атрибуту Pandas .columns
Загрузка примера DataFrame
Как получить имена столбцов из DataFrame Pandas
Как получить список имён столбцов из CSV-файла
Как проверить существование столбца в DataFrame
Как посчитать количество столбцов в DataFrame
Как получить словарь имён столбцов и их типов данных
Как получить имена столбцов определённого типа данных
Как получить имя столбца по индексу
Как получить имена столбцов в многоуровневом (MultiIndex) DataFrame
Как получить имена столбцов, удовлетворяющих определённому условию
Как получить имена столбцов, содержащих пропущенные значения
Как получить имена столбцов с дублирующимися значениями
Как получить алфавитный список имён столбцов
Как получить имена столбцов, начинающихся с определённой буквы
Часто задаваемые вопросы
Заключение
Дополнительные ресурсы