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中的groups已经给您groupby

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)}

按键与熊猫系列分组并导出到_dict（）

1 个答案: