python - 定义大量列表的便捷方法？

Question

对于我当前的代码，我需要定义很多列表并将它们返回到字典中。

仅此一项就占用了约 100 行代码。它有效，但我很好奇是否有更好的方法来编写此代码

list1 = []
list2 = []
list3 = []
.
.
.

return {"list1": list1,
"list2": list2,
"list3": list3,
...
}

我首先定义 ~40 个列表，在它们后面附加数据，并在进行一些额外计算后在字典中返回 ~60-70 个列表(稍后我使用这些数据创建数据帧)。

这使得代码变得非常庞大，那么有没有办法“美化”这段代码呢？

Answer 1

根据您这样做的原因，您也许可以使用 collections 中一个非常有用的类，称为 defaultdict。该对象的操作类似于字典，但当您调用不存在的键时，它也有一个默认返回值。

这意味着您可以将代码替换为:

from collections import defaultdict

data = defaultdict(list)

这意味着每当您访问一个值时，如果之前不存在该值，它将创建一个空列表。这使您可以像以前一样附加到列表中:

data['list1'].append('your_value_here')