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

Question

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

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-код']) дает

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

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

Answer 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]

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