我正在尝试使用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] 删除。
我来说两句