使用python中的標頭將兩個單獨的列表寫入csv

Question

list1 = [1,2,4,5,7,8,9]  corresponds to ID
list2 = [3,2,1,1,1,2,3]  corresponds to relevance

我有兩個不同的列表，分別是list1，list2和數值。 我想將兩者都寫入標頭為[“ ID”，“ Relevance”]的csv文件中。 現在，我想在標題“ ID”中寫入list1的值，在標題“相關性”中寫入list2的值。 我在下面的代碼中嘗試寫兩個列表，但沒有給出期望的輸出。

我希望輸出采用以下示例格式：

ID    Relevance
1         3
3         2
4         2
6         1
7         3
10        1

碼：

with open('results_cosine11.csv', 'w', encoding="ISO-8859-1", newline='') as myfile:
    wr = csv.writer(myfile)
    for value in list1,list2:
        wr.writerow([value])

Answer 1

您正在尋找的是zip() 。 這將起作用：

with open('results_cosine11.csv', 'w', encoding="ISO-8859-1", newline='') as myfile:
    wr = csv.writer(myfile)
    for value in zip(list1,list2):
        wr.writerow(value)

或者，更有效地：

with open('results_cosine11.csv', 'w', encoding="ISO-8859-1", newline='') as myfile:
    wr = csv.writer(myfile)
    wr.writerows(zip(list1,list2))

---

為了也寫標題：

with open('results_cosine11.csv', 'w', encoding="ISO-8859-1", newline='') as myfile:
    wr = csv.writer(myfile)
    wr.writerow(("ID", "Relevance"))
    wr.writerows(zip(list1,list2))

Answer 2

嘗試

a= zip(list1,list2)
for id,relevance in a:
   #... do  whatever