与熊猫系列按键分组并导出 to_dict()

Question

我有一本像这样的字典：

d = {1:0, 2:0, 3:1, 4:0, 5:2, 6:1, 7:2, 8:0} 

我想按我得到的 .keys() 分组：

pandas_ordered = { 0:[1,2,4,8], 1:[3,6], 2:[5,7] }

但是对于

pd.Series(list(d.values())).groupby(list(partition.keys())).to_dict()

下面是一个例子：

# Example:
import pandas as pd

d = {1:0, 2:0, 3:1, 4:0, 5:2, 6:1, 7:2, 8:0} 


def pandas_groupby(dictionary):
   values = list(dictionary.values())
   keys = list(dictionary.keys())
   return pd.Series(values).groupby(keys).to_dict()


pandas_groupby(d)

以上代码产生错误：

AttributeError：无法访问的可调用属性“to_dict” 'SeriesGroupBy' 对象，尝试使用 'apply' 方法

关于如何做到这一点的任何想法？

Answer 1

您的dict 已经由您的groupby 中的groups 给出

d = {1:0, 2:0, 3:1, 4:0, 5:2, 6:1, 7:2, 8:0} 
s = pd.Series(d)
s.groupby(s).groups

{0: Int64Index([1, 2, 4, 8], dtype='int64'),
 1: Int64Index([3, 6], dtype='int64'),
 2: Int64Index([5, 7], dtype='int64')}

当然，总是可以agg 和自定义

s.groupby(s).agg(lambda x: tuple(x.index)).to_dict()

{0: (1, 2, 4, 8), 1: (3, 6), 2: (5, 7)}