Pandas groupby(字典)没有返回预期的结果

时间:2013-06-06 17:27:24

标签: pandas

我正在尝试将以下数据分组:

>>> a=[{'A': 1, 'B': 2, 'C': 3, 'D':4, 'E':5, 'F':6},{'A': 2, 'B': 3, 'C': 4, 'D':5, 'E':6, 'F':7},{'A': 3, 'B': 4, 'C': 5, 'D':6, 'E':7, 'F':8}]
>>> df = pd.DataFrame(a)
>>> df
   A  B  C  D  E  F
0  1  2  3  4  5  6
1  2  3  4  5  6  7
2  3  4  5  6  7  8

使用以下词典:

dict={'A':1,'B':1,'C':1,'D':2,'E':2,'F':2}

这样

df.groupby(dict).groups

将输出

{1:['A','B','C'],2:['D','E','F']}

1 个答案:

答案 0 :(得分:2)

需要将axis参数添加到groupby:

>>> grouped = df.groupby(groupDict,axis=1)
>>> grouped.groups
{1: ['A', 'B', 'C'], 2: ['D', 'E', 'F']}