如何提取pandas.DataFrame数据表中某一列的类别

在数据处理中遇到一个问题：如何提取数据表中某一列的类别，也就是汇总多少种类。

问题很简单，有以下两种方法，个人推荐第2种。

1.直接对数据表进行透视，然后取出类别的一列

df_pivot = df.pivot_table(index = 'Catagory', values = 'Val')

Cata_list = df_pivot.index.tolist()

这里如果数据量不多的情况下，没有问题。但数据一多就会造成资源浪费，因为透视默认必须对一个'Val'列进行计算，实际上这里没有必要（以上默认计算了mean())。

同时，如果某一列如果没有数据Nan，也会报错。

2.直接分组，提取组字典的key值

df_group = df.groupby(by = 'Catagory')

Cata_list = list(df_group.groups.keys())

这里是按'Catagory'进行分组，用groups属性返回组字典，再用字典中keys()方法提取类别，再转化为list类型。