使用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

I have two different lists say list1,list2 with numerical values in it. I want to write both to a csv file with headers ["ID","Relevance"]. Now I want to write values of list1 in header "ID" and list2 in header "Relevance". I have written below code trying to write both lists but it doesnt give desired output.

I want output to be in below sample format:

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

code:

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

What you're looking for is zip() . This will work:

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)

Or, more efficiently:

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

---

In order to write the header too:

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

try

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