将嵌套在两个字典下的列表转换为DataFrame

威尼斯人

我正在尝试使用Pandas在Python中创建一个数据框架,该数据框架涉及嵌套字典和列表列表。我浏览了有关转换嵌套字典的其他问题,但找不到足够的答案。

我有一本字典,例如,一本活动簿,记录了课外学校的课程。在这种情况下,有两个课程,每个课程都是嵌套在活动簿字典下的自己的字典。每个课程词典都包含按月组织的每个人的活动列表。每个月执行一项活动的学生数量是可变的,但结构始终是Student-Activity-Minutes。例如:

activity_dict = {

'lesson1' : {  'january' : [['Todd', 'Running', 30],['Christy', 'Studying', 25],['Alex','Soccer', 10]],
               'february' : [['Jim', 'Bobsledding', 5],['Frank', 'Jogging',8]]},

'lesson2' : {'february' : [['Todd', 'Running', 18],['John', 'Studying', 3],['Don','Soccer', 40]],
              'march' : [['Tom', 'Bobsledding', 10],['Sam', 'Yoga', 42]],
              'april' : [['Julie', 'Biking', 20],['Chris', 'Baseball', 10]]}
}  

我正在尝试获取每个学生的活动的输出,ColA =课号,ColB =月,ColC =学生,ColD =活动,ColE =分钟。样本输出为:

Lesson # Month Student Activity Minutes
Lesson 1 February Jim Bobsledding 5
Lesson 1 February Frank Jogging 8
Lesson 2 February Todd Running 18

我找到了一种创建列C到E的数据框的方法,但是我无法包含列A和B。

我现在的代码如下:

import pandas

activity_log = []

for lesson, all_activities in activity_dict.items():
    for month, month_activities in all_activities.items():
        activity_log.append(pandas.DataFrame(month_activities))

如何更新它以将字典键(课程和月份)作为A列和B列?我不确定将列表列表更改为字典是否有帮助,但是我将其保留为列表,因为这是我接收数据的方式。

忘了它

使用列表推导将列表列表的字典转换为列表列表:

In [99]: [(lesson, month, name, activity, minutes) 
          for lesson, dct in activity_dict.items() 
          for month, vals in dct.items() 
          for name, activity, minutes in vals]
Out[99]: 
[('lesson2', 'april', 'Julie', 'Biking', 20),
 ('lesson2', 'april', 'Chris', 'Baseball', 10),
 ('lesson2', 'february', 'Todd', 'Running', 18),
 ('lesson2', 'february', 'John', 'Studying', 3),
 ('lesson2', 'february', 'Don', 'Soccer', 40),
 ('lesson2', 'march', 'Tom', 'Bobsledding', 10),
 ('lesson2', 'march', 'Sam', 'Yoga', 42),
 ('lesson1', 'january', 'Todd', 'Running', 30),
 ('lesson1', 'january', 'Christy', 'Studying', 25),
 ('lesson1', 'january', 'Alex', 'Soccer', 10),
 ('lesson1', 'february', 'Jim', 'Bobsledding', 5),
 ('lesson1', 'february', 'Frank', 'Jogging', 8)]

然后使用pd.DataFrame来从列表列表构建DataFrame:

In [98]: pd.DataFrame([(lesson, month, name, activity, minutes)
                       for lesson, dct in activity_dict.items() 
                       for month, vals in dct.items() 
                       for name, activity, minutes in vals], 
             columns=['Lesson', 'Month', 'Name', 'Activity', 'Minutes'])
Out[98]: 
     Lesson     Month     Name     Activity  Minutes
0   lesson2     april    Julie       Biking       20
1   lesson2     april    Chris     Baseball       10
2   lesson2  february     Todd      Running       18
3   lesson2  february     John     Studying        3
4   lesson2  february      Don       Soccer       40
5   lesson2     march      Tom  Bobsledding       10
6   lesson2     march      Sam         Yoga       42
7   lesson1   january     Todd      Running       30
8   lesson1   january  Christy     Studying       25
9   lesson1   january     Alex       Soccer       10
10  lesson1  february      Jim  Bobsledding        5
11  lesson1  february    Frank      Jogging        8

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章