Python写入CSV时writerow()和writerows()函数的使用示例

2024-01-06 03:02:36

用 Python 的 writerow() 和 writerows() 函数编写 CSV 文件

CSV（逗号分隔值）是一种常见的文本文件格式，用于存储表格数据。它易于使用，可以被广泛的应用程序读取。Python 提供了 csv 模块，其中包含 writerow() 和 writerows() 等函数，可轻松地将数据写入 CSV 文件。

writerow() 函数

writerow() 函数将一行数据写入 CSV 文件。该函数接受一个列表作为参数，该列表包含要写入的各列的值。它一次只能写入一行数据。以下是使用 writerow() 函数的示例：

import csv

with open('data.csv', 'w') as f:
    csv_writer = csv.writer(f)
    csv_writer.writerow(['Name', 'Age', 'City'])

writerows() 函数

writerows() 函数将多行数据写入 CSV 文件。该函数接受一个列表作为参数，该列表包含要写入的数据行。每个行又是一个列表，包含要写入的各列的值。它一次可以写入多行数据。以下是使用 writerows() 函数的示例：

import csv

with open('data.csv', 'w') as f:
    csv_writer = csv.writer(f)
    csv_writer.writerows([
        ['John', 30, 'New York'],
        ['Mary', 25, 'Los Angeles'],
        ['Bob', 40, 'Chicago']
    ])

writerow() 和 writerows() 的比较

虽然 writerow() 和 writerows() 都用于将数据写入 CSV 文件，但它们之间存在一些关键差异：

处理的数据量： writerow() 只能一次写入一行数据，而 writerows() 可以一次写入多行数据。
效率： writerow() 比 writerows() 更高效，因为 writerow() 只需写入一行数据，而 writerows() 需要写入多行数据。
参数： writerow() 接受一个包含要写入各列值的数据行列表作为参数，而 writerows() 接受一个包含数据行列表的列表作为参数。

何种情况使用哪种函数

在选择使用 writerow() 还是 writerows() 时，需要考虑以下因素：

要写入的数据量： 如果您要写入少量数据，可以使用 writerow()。但是，如果您要写入大量数据，则应使用 writerows() 以提高效率。
写入数据的方式： 如果您要逐行写入数据，可以使用 writerow()。但是，如果您要将数据存储在列表中，则应使用 writerows()。
效率： 如果写入效率至关重要，可以使用 writerow()。