Python 3.6。如何在for循环中迭代多个列表并使用DictWriter()将它们写入CSV文件

时间:2017-11-22 18:01:54

标签: python list csv format export-to-csv

我正在尝试迭代多个列表并将其保存为CSV文件以在电子表格中打开,第一个列表将其值添加到第1,2和3列,第二个列表将其添加到&# 39; s值到第4,5,6列。我反而得到列表1的第1行len(列表2)的次数,然后列表2的第二行相同,依此类推。列表2正常打印出来,然后在列表1的第二行最终打印后再次打印。希望我能够很好地解释这一点。

ixl_file = open('ixalan.txt', 'r')
ixl_card = []
ixl_price = []
ixl_quantity = []
for line in ixl_file:
    info = line.split("|")
    ixl_card.append(info[0])
    ixl_price.append(info[1])
    ixl_quantity.append(info[2])
ixl_file.close()

kal_file = open('kaladesh.txt', 'r')
kal_card = []
kal_price = []
kal_quantity = []
for line in kal_file:
    info = line.split("|")
    kal_card.append(info[0])
    kal_price.append(info[1])
    kal_quantity.append(info[2])
kal_file.close()

with open('card_kingdom.csv', 'w', newline='') as ixl:
        fieldnames = ['Col 1', 'Col 2', 'Col 3', 'Col 4', 'Col 5', 'Col 6']
    writer = csv.DictWriter(ixl, fieldnames=fieldnames)
    writer.writeheader()

"""以下是我无法以所需格式添加这两个列表的地方"""

    for a,b in itertools.product(range(len(ixl_card)), range(len(kal_card))):
        writer.writerow({'Col 1': ixl_card[a],
                         'Col 2': ixl_price[a],
                         'Col 3': ixl_quantity[a],
                         'Col 4': kal_card[b],
                         'Col 5': kal_price[b],
                         'Col 6': kal_quantity[b]
                       })

非常感谢您的帮助!如果您需要其余代码,请告诉我。谢谢!

1 个答案:

答案 0 :(得分:1)

这可以使用普通的CSV编写器完成,因为所有列的顺序都是相同的:

import csv    

with open('ixalan.txt', newline='') as f_ixalan:
    ixalan = list(csv.reader(f_ixalan, delimiter='|'))

with open('kaladesh.txt', newline='') as f_kaladesh:
    kaladesh = list(csv.reader(f_kaladesh, delimiter='|'))

with open('card_kingdom.csv', 'w', newline='') as f_output:
    csv_output = csv.writer(f_output)
    csv_output.writerow(['Col 1', 'Col 2', 'Col 3', 'Col 4', 'Col 5', 'Col 6'])

    for p1, p2 in zip(ixalan, kaladesh):
        csv_output.writerow(p1 + p2)

此外,通过为分隔符指定.txt,可以使用csv.reader()读取您的两个|文件。使用zip()从两个输入列表中的每一个中获取一行,然后将这些列合并并写入输出CSV。