Groupby到CSV文件

Question

我有一个groupby，我想仅将groupby（无计数，总和等）转换为Dataframe，然后将其导出为CSV。

从本质上讲，这只是查找几列数据中的所有唯一组合。 如果在不使用groupby的情况下更容易做，那么我也可以。

我尝试使用reset_index（）并收到以下消息。

AttributeError: Cannot access callable attribute 'reset_index' of 'DataFrameGroupBy' objects, try using the 'apply' method

data = pd.DataFrame([['a','z'],['a','y'],['b','y'],['a','y']],columns=['one','two'])
gb = data.groupby(['one','two'])['one'].count()

我只想输出具有“一”和“二”的唯一组合的三行数据框。

Answer 1

如果只需要两列的唯一组合，请在这些列上调用drop_duplicates() ：

unique_combs = data[['one', 'two']].drop_duplicates()

unique_combs
Out[32]: 
  one two
0   a   z
1   a   y
2   b   y