将 Python 列表保存为 CSV 文件

在 python 中，将列表数据导出到 csv 文件是一个常见的任务。然而，初学者经常会遇到一个问题：当使用 csv.writerows() 方法时，列表中的每个字符串元素会被错误地拆分成单个字符，并分别写入不同的列。本文将详细介绍如何避免这个问题，并提供正确的代码示例，确保列表中的每个元素都作为单独的行写入 csv 文件。

使用 csv 模块正确写入列表数据

Python 的 csv 模块提供了方便的方法来处理 CSV 文件的读写操作。要将列表中的数据写入 CSV 文件，可以使用 csv.writer 对象和其 writerow() 或 writerows() 方法。关键在于理解这些方法对输入数据的期望格式。

writerow(row) 方法接受一个可迭代对象（如列表或元组），并将该可迭代对象中的每个元素写入 CSV 文件的一行，每个元素占据一个单元格。

writerows(rows) 方法接受一个可迭代对象，该可迭代对象中的每个元素本身也是一个可迭代对象（通常是列表或元组）。writerows() 方法会将每个内部的可迭代对象写入 CSV 文件的一行。

常见错误：字符分隔

当直接将一个字符串列表传递给 writerows() 方法时，Python 会将每个字符串视为一个字符列表，导致每个字符被写入不同的列。例如：

立即学习“Python免费学习笔记（深入）”；

import csv

links = ['https://www.example.com/page1', 'https://www.example.com/page2']

with open('links.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerows(links)

上述代码会导致 links.csv 文件中的每一行都只包含一个字符，而不是预期的完整 URL。

正确方法：将每个元素包装在列表中

为了解决这个问题，需要确保传递给 writerows() 方法的每个元素都是一个包含单个元素的列表。可以使用列表推导式或生成器表达式来实现这一点。

存了个图

视频图片解析/字幕/剪辑，视频高清保存/图片源图提取

17

查看详情

方法一：使用列表推导式

import csv

links = ['https://www.example.com/page1', 'https://www.example.com/page2']

with open('links.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerows([[link] for link in links])

方法二：使用生成器表达式

import csv

links = ['https://www.example.com/page1', 'https://www.example.com/page2']

with open('links.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerows([link] for link in links)

这两种方法都会将 links 列表中的每个 URL 包装在一个单独的列表中，然后将这些列表传递给 writerows() 方法，从而确保每个 URL 都被写入 CSV 文件的一行。

使用 writerow() 方法

另一种可行的方法是使用 writerow() 方法，每次写入一行数据。

import csv

links = ['https://www.example.com/page1', 'https://www.example.com/page2']

with open('links.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    for link in links:
        writer.writerow([link])

这个方法循环遍历 links 列表，并将每个 link 包装在一个列表中，然后使用 writerow() 方法将其写入 CSV 文件。

注意事项

• newline='' 参数： 在打开文件时，newline='' 参数可以避免在 Windows 系统上出现额外的空行。
• 编码问题： 如果数据包含非 ASCII 字符，可能需要指定编码方式，例如 encoding='utf-8'。
• 分隔符： 默认情况下，CSV 文件的分隔符是逗号。可以使用 delimiter 参数指定其他分隔符，例如 csv.writer(f, delimiter=';')。

总结

将 Python 列表数据正确写入 CSV 文件需要理解 csv 模块中 writerows() 和 writerow() 方法的用法。通过将列表中的每个元素包装在另一个列表中，可以避免字符分隔的问题，确保数据以预期的格式写入 CSV 文件。根据具体情况选择合适的方法，并注意编码和分隔符等细节，可以有效地处理 CSV 文件的读写操作。

以上就是将 Python 列表保存为 CSV 文件的详细内容，更多请关注php中文网其它相关文章！