数据框列名比较

Question

数据框国家/地区在下面的列中

snd x < snd y

如果Country.columns包含“ oecd-code”列，我想执行一些操作

我的代码：

otherwise

Country.columns.isin（['oecd-code']）给出

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

如果具有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]

编辑：如果仅想检查'oecd-code'是否为Country的列，则只需检查'oecd-code' in Country.columns或实际上为'oecd-code' in Country，或者如果您已经拥有您的布尔数组，请使用.any()检查是否有任何值是True。