获取所有分类列并排除一个特定列
获取所有分类列并排除一个特定列
提问于 2019-02-05 17:32:06
我有一个名为data的数据框。我希望选择除名为test的列之外的所有对象和分类数据类型。
我有以下代码，但它选择了所有列
| testid | color | age | mark |
+--------+-------+-----+------+
| a | red | 20 | pass |
| d | green | 30 | fail |
| c | blue | 40 | pass |
data = pd.DataFrame([['a', 'red', 20, 'pass'], ['d', 'green', 30, 'fail'], ['c', 'blue', 40, 'pass']], columns=['testid', 'color', 'age', 'mark'])
for i in data.select_dtypes(include=['object','category']).columns:
data[i]=le.fit_transform(data[i])我想转换除testid之外的所有行
回答 3
Stack Overflow用户
发布于 2019-02-05 17:44:09
您可以使用pd.Index.difference
cols = data.select_dtypes(include=['O', 'category']).columns.difference(['testid'])
for i in cols:
data[i] = le.fit_transform(data[i])Stack Overflow用户
发布于 2019-02-05 17:35:10
您可以将if语句放在下面这样：
for i in data.select_dtypes(include=['object','category']).columns:
if i == 'testid':
continue
data[i]=le.fit_transform(data[i])Stack Overflow用户
发布于 2019-02-05 22:11:07
另一个简短的解决方案：
cols = data.dtypes.isin(['O','category']).drop('testid').index
for i in cols:
data[i] = le.fit_transform(data[i])页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/54531253
复制相关文章
相似问题
腾讯云开发者
