Python CSV 数据分割：根据第一列值将 CSV 文件分成两个文件

以下是一个 Python 代码示例，演示如何读取 CSV 文件并将其分成两个文件，一个包含第一列为'1' 的行，另一个包含第一列为'3' 的行：

import csv

# 打开 CSV 文件并读取数据
with open('input.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    # 创建两个输出文件
    with open('output1.csv', 'w', newline='') as outputfile1, open('output3.csv', 'w', newline='') as outputfile3:
        writer1 = csv.writer(outputfile1)
        writer3 = csv.writer(outputfile3)
        # 遍历 CSV 文件中的每一行
        for row in reader:
            # 如果第一列是'1'，则将行写入第一个输出文件
            if row[0] == '1':
                writer1.writerow(row)
            # 如果第一列是'3'，则将行写入第二个输出文件
            elif row[0] == '3':
                writer3.writerow(row)

在此示例中，我们首先打开 CSV 输入文件（命名为'input.csv'），然后使用 Python 的 CSV 模块读取数据。接下来，我们创建两个输出文件（命名为'output1.csv' 和 'output3.csv'），并使用 CSV 模块的 writer 方法创建两个 CSV 写入器，一个用于第一列为'1' 的行，另一个用于第一列为'3' 的行。

然后，我们使用一个 for 循环遍历输入 CSV 文件中的每一行。在循环中，我们检查第一列的值，如果是'1'，则将该行写入第一个输出文件；如果是'3'，则将该行写入第二个输出文件。

最后，我们使用 Python 的'with' 语句来自动关闭所有打开的文件，以确保数据正确地写入磁盘。