在数据集中查找对象

【问题标题】：Finding objects in dataset在数据集中查找对象
【发布时间】：2023-02-10 15:00:56
【问题描述】：

我试图了解如何在数据集中查找对象列，但我遇到了一个小问题

s = (X_train.dtypes == 'object')
object_cols = list(s[s].index)

print("Categorical variables:")
print(object_cols)

s[s].index 在这里做什么？

【问题讨论】：

标签： python pandas categorical-data

【解决方案1】：

如果 s 是布尔掩码，则通过此掩码从 X_train.columns 中过滤 s.index 并将值转换为 list。

所以它的工作方式如下：

X_train = pd.DataFrame({'col1':list('abc'),
                        'col2':list('ger'),
                        'col3':[4,5,6]})

s = (X_train.dtypes == 'object')
print(s)
col1     True
col2     True
col3    False
dtype: bool

object_cols = list(s[s].index)

print("Categorical variables:")
print(object_cols)
['col1', 'col2']

object_cols = list(X_train.columns[s])
print("Categorical variables:")
print(object_cols)
['col1', 'col2']

【讨论】：