pandas pivot table 转换为 defaultdict(list)

Question

我使用 pandas 枢轴 table。我需要将数据从 pivot table 转换为 defaultdict(list)

这是我要听写的枢轴 table 数据 (df.to_dict()):

{
    'count': {
        ('formation1', 'position1'): 2,
        ('formation1', 'position2'): 1
    },
    'personData.employeeContract.edges': {
        ('formation1','position1'): 1,
        ('formation1', 'postition2'): 0
    },
    'total_count': {
        ('formation1', 'position1'): 2,
        ('formation1', 'position2'): 1
    },
    'count_with_contract': {
        ('formation1', 'position1'): 1,
        ('formation1', 'position2'): 0
    },
    'percent': {
        ('formation1', 'position1'): 0.5,
        ('formation1', 'position2'): 0.0
    }
}

我需要将上面的数据传输到下面：

{
    'formation1': [{
            'position1': {
                'total_count': 2.0
                'count_with_contract': 1.0
                'percent': 0.0
                
            } {
                'position2': {
                    'total_count': 1.0
                    'count_with_contract': 0.0
                    'percent': 0.0
                }
            ]

        }
    }

我该怎么做？

Answer 1

首先过滤需要的列名称，然后在字典理解中创建嵌套字典：

df1 = df[['total_count','count_with_contract','percent']]

d = {i:[g.reset_index(level=0, drop=True).to_dict('index')] for i,g in df1.groupby(level=0)}
print (d)
{'formation1':[{'position1': {'total_count': 2, 'count_with_contract': 1, 'percent': 0.5}, 
                'position2': {'total_count': 1, 'count_with_contract': 0, 'percent': 0.0}}]}

defaultdict的解决方案（输出有点不同）：

from collections import defaultdict

df1 = df[['total_count','count_with_contract','percent']]
print (df1)

d = defaultdict(list)
for (f, pos), x in df1.T.items():
    d[f].append({pos: x.to_dict()})
    
print (d)
defaultdict(<class 'list'>, 
{'formation1': [{'position1': {'total_count': 2.0, 'count_with_contract': 1.0, 'percent': 0.5}}, 
                {'position2': {'total_count': 1.0, 'count_with_contract': 0.0, 'percent': 0.0}}]})

pandas pivot table 转换为 defaultdict(list)

pandas pivot table convert to defaultdict(list)

python

pandas

defaultdict