python - 增加不存在索引的列表索引

Question

我正在尝试迭代数据集的行，并增加新创建的列表的索引。我知道我目前正在尝试增加一个不存在的索引，但我非常感谢任何试图完成类似事情的帮助。

这是我的代码片段:

attack_year_type = []
year_type_counter = []
for _, event in main_df.iterrows():
   attack_year_type.append((event['iyear'], event['attacktype1_txt']))
   year_type_counter[int(event['iyear'])][event['attacktype1_txt']] += 1

Answer 1

我认为您最好将 year_type_counter 作为 dict 的 dict 。在这种情况下，您可以使用 defaultdict 来完成您想要的操作。

from collections import defaultdict

year_type_counter = defaultdict(lambda: defaultdict(int))
attack_year_type = []

for _, event in main_df.iterrows():
    attack_year_type.append((event['iyear'], event['attacktype1_txt']))
    year_type_counter[int(event['iyear'])][event['attacktype1_txt']] += 1