从火车数据集中获取分类特征时出错
error in getting categorial features from train dataset
我的火车数据看起来像这样：
train data
为了从中提取分类特征，我运行遵循以下代码
categorial=[c for c in train.columns if train.columns(c).dtype in ['object'] ]
但我收到错误消息：
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
<ipython-input-31-31eb7ac47e21> in <module>
----> 1 categorial=[c for c in train.columns if train.columns[c].dtype in ['object'] ]
<ipython-input-31-31eb7ac47e21> in <listcomp>(.0)
----> 1 categorial=[c for c in train.columns if train.columns[c].dtype in ['object'] ]
/opt/conda/lib/python3.7/site-packages/pandas/core/indexes/base.py in __getitem__(self, key)
4295 if is_scalar(key):
4296 key = com.cast_scalar_indexer(key, warn_float=True)
-> 4297 return getitem(key)
4298
4299 if isinstance(key, slice):
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices
可能的解决方案是什么？
将此用于select 'object' 类型变量-
categorical = train.select_dtypes('object')
如果你只想要变量名 -
categorical_cols = train.select_dtypes('object').columns.tolist()
我的火车数据看起来像这样： train data
为了从中提取分类特征，我运行遵循以下代码
categorial=[c for c in train.columns if train.columns(c).dtype in ['object'] ]
但我收到错误消息：
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
<ipython-input-31-31eb7ac47e21> in <module>
----> 1 categorial=[c for c in train.columns if train.columns[c].dtype in ['object'] ]
<ipython-input-31-31eb7ac47e21> in <listcomp>(.0)
----> 1 categorial=[c for c in train.columns if train.columns[c].dtype in ['object'] ]
/opt/conda/lib/python3.7/site-packages/pandas/core/indexes/base.py in __getitem__(self, key)
4295 if is_scalar(key):
4296 key = com.cast_scalar_indexer(key, warn_float=True)
-> 4297 return getitem(key)
4298
4299 if isinstance(key, slice):
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices
可能的解决方案是什么？
将此用于select 'object' 类型变量-
categorical = train.select_dtypes('object')
如果你只想要变量名 -
categorical_cols = train.select_dtypes('object').columns.tolist()