我有从 csv 文件中读取的旅行日记的数据。我将它设置为带有一堆列表的字典。例如:
print diary['ID'][1] gives 123456789
print diary['TravelReferenceDay'][1] gives 1 for a Monday
我想根据日期从数组中随机选择一个 ID,例如:
random.choice(diary['ID']) if diary['TravelReferenceDay'] == 1
我可以在 csv 文件中按 TravelReferenceDay 排列数据。我曾尝试使用 groupby 方法来拆分数组:
groups = []
uniquekeys = []
for k, g in groupby(diary, diary['TravelReferenceDay']):
groups.append(list(g)) # Store group iterator as a list
uniquekeys.append(k)
但这给出了错误:
TypeError: 'list' object is not callable
你能建议一种方法来实现这一目标吗?谢谢。