python - 分组 Pandas 数据框中的嵌套字典

Question

我有一个像这样的 Pandas 数据框:

    Tail Number | flight route | digit
    _______________
    001C     | (VB, MI)   | 1
             | (CC, SK)   | 2
             | (KF, KC)   | 1
    004N     | (AZ, AL)   | 2
             | (AU, NY)   | 3
    005F     | (ALB, TPA) | 5
             | (ORD, JAC) | 2

我感兴趣的变量是尾号、飞行路线(元组)和数字(整数值列)。

我想用它制作一个嵌套字典，以尾号为键，另一个内部字典，其中键是航线，值是数字。理想情况下，嵌套字典如下所示:

d = {001C :{  {(VB, MI): 1}, {(CC, SK): 2},  {(KF, KC):1} },
    004N :{    {(AZ, AL): 2}, {(AU, NY):3},
    005F :{  {(ALB, TPA):5},  {(ORD, JAC):2},  ... }

你能帮帮我吗？

Answer 1

一种方法是使用 groupby 和字典理解

{k: dict(zip(v['flight route'], v['digit'])) for k, v in df.groupby('Tail Number')}

Out[209]:
{'001C': {('VB', 'MI'): 1, ('CC', 'SK'): 2, ('KF', 'KC'): 1},
 '004N': {('AZ', 'AL'): 2, ('AU', 'NY'): 3},
 '005F': {('ALB', 'TPA'): 5, ('ORD', 'JAC'): 2}}