使用多个键的默认字典集合

Question

我正在用 Python 的集合 defaultdict 来读取一个很大的 CSV 文件。我的 CSV 文件的表头大概是这样的：

main_id, id1, value_x, id2, id3, id4, value_y, value_z, value0

我可以一行一行地读取这个文件，并把 main_id 作为默认的键。请问有没有办法使用多个键，这样字典就可以像 dict[main_id,id1,id2,id3,id4] 这样？

Answer 1

使用一个包含元素的 tuple。这样你就可以得到

data = dict()
data[(main_id,id1,id2,id3,id4)]

或者直接使用

data = dict()
data[main_id,id1,id2,id3,id4]  # defaults to a tuple

这正是你想要实现的效果。