Сравнение имен столбцов фрейма данных

В стране фрейма данных есть столбцы ниже

Index(['hasData', 'key', 'level', 'name', 'id', 'oecd-code', 'regionid'], dtype='object')

Я хочу выполнить некоторые действия, если Country.columns содержит столбец oecd-code

Мой код:

if Country.columns.isin(['oecd-code']):
      #Some actions

Country.columns.isin (['oecd-code']) дает

array([False, False, False, False, False,  True, False], dtype=bool)

Как выполнить фильтрацию из массива, если он имеет значение True?

python python-3.x dataframe

27.10.2018 14:01

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Если вы настаиваете на использовании .isin, вы можете:

select_cols = ['oecd-code']
col_mask = Country.columns.isin(select_cols)
Country[Country.columns[col_mask]]

или

Country.loc[:, col_mask]

Однако вы можете просто сделать следующее.

Country[select_cols]

Обновлено: Если вы хотите только проверить, является ли 'oecd-code' столбцом Country, просто проверьте 'oecd-code' in Country.columns или фактически 'oecd-code' in Country, или, если у вас уже есть логический массив, используйте .any(), чтобы проверить, является ли какое-либо из значений True.

Я просто хочу проверить, существует ли столбец «oecd-code» в Country.columns или нет?

— 27.10.2018 14:20

27.10.2018 14:12