我不知道逻辑上遍历列表并获取每个特定索引信息并将其添加到lis的最佳方法

Question

我已经编写了这段代码，但现在它是用特定索引硬编码的。我想遍历一个数据帧，对于列表中的每个索引，我想得到总和

    sums = []
    test = []
    for index, row in df.iterrows():
         if row['subject_id'] == subject_id_list[0]:
             test.append(row['answer'])
             test = list(map(int, test))
    test_sum = sum(test)
    sums.append(test_sum)

如果列表中有5个科目，我需要每个科目的总和，并将其放入所有总和的新列表中

Answer 1

你能再加一个for循环吗？大概

for index, row in df.iterrows():
   for item in subject_id_list:
        if row['subject_id'] == item:
             test.append(row['answer'])
test_sum = sum(test)
sums.append(test_sum)