python - 如何将多个列表中的元素写入 csv，其中每个元素属于每一行？

Question

请帮我解决有关 csv 文件的问题。我有 2 个列表，我想将它们写入 csv 文件，如下所示:

X = [12,423,56]
Y = [35,78,321]

csv 文件还包含一个自动生成的索引列“No”:

No,X,Y
1,12,35
2,423,78
3,56,321

Answer 1

您可以使用 csv 来执行此操作标准库中的模块，以及 zip和 enumerate内置函数。

with open('myfile.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    # Write column headers
    writer.writerow(['No', 'X', 'Y'])
    for i, (a, b) in enumerate(zip(X,Y), start=1):
        writer.writerow([i, a, b])

输出:

No,X,Y
1,12,35
2,423,78
3,56,321

csv 模块提供了读取和写入 csv 文件的工具。

enumerate 输出循环每次迭代的索引；这里我们将其配置为使用 1 作为起始值 - 默认情况下它从零开始。

zip 接受一个或多个可迭代对象作为输入，循环遍历每个可迭代对象，并每次从可迭代对象的后续位置生成元素:

• 第一次迭代:从每个可迭代中产生项目 0
• 第二次迭代:从每个可迭代中产生第 1 项

等等。

如果可迭代对象的长度不一样。一旦最短的可迭代对象被消耗完，zip 就会停止生成。