python - 检查 DataFrame 中的哪些列是分类的

Question

我是 Pandas 的新手...当我不手动指定时，我想要一种简单而通用的方法来查找我的 DataFrame 中哪些列是 categorical每个列类型，不像 this SO question . df 使用以下命令创建:

import pandas as pd
df = pd.read_csv("test.csv", header=None)

例如

           0         1         2         3        4
0   1.539240  0.423437 -0.687014   Chicago   Safari
1   0.815336  0.913623  1.800160    Boston   Safari
2   0.821214 -0.824839  0.483724  New York   Safari

.

更新 (2018/02/04) 问题假设数字列不是分类的，@Zero 的 accepted answer solves this .

小心 - 正如@Sagarkar 的评论指出的那样，这并不总是正确的。 困难在于数据类型和分类/有序/名义类型是正交概念，因此它们之间的映射并不简单。 @杰夫的answer下面指定了实现手动映射的精确方式。

Answer 1

您可以使用 df._get_numeric_data() 获取数字列，然后找出分类列

In [66]: cols = df.columns

In [67]: num_cols = df._get_numeric_data().columns

In [68]: num_cols
Out[68]: Index([u'0', u'1', u'2'], dtype='object')

In [69]: list(set(cols) - set(num_cols))
Out[69]: ['3', '4']