将特定行从一个 CSV 文件添加到另一个文件并应用特定条件

Question

我有一个包含 0 和 1 值的“标志”列的 CSV 文件。我的目标是将所有具有 0 值的行移动到另一个 CSV 文件。该脚本将安排为每小时运行一次，并将具有“0”值的行移动到另一个文件。

到目前为止，我编写了以下代码：

with open("path/to/my/input/file.csv", "rt", encoding="utf8") as f:
reader = csv.DictReader(f, delimiter=',')
with open("/path/to/my/output/file.csv", "a+", encoding="utf8") as f_out:
    writer = csv.DictWriter(f_out, fieldnames=reader.fieldnames, delimiter=",")
    writer.writeheader()
    for row in reader:
        if row['flag'] == '0':
            writer.writerow(row)

借助下面的@Raghvendra 帮助，通过在我的代码中添加“a+”，我可以将行添加到我的 output.csv 文件中。但是，每次脚本运行时，它都会将标题行添加到我的输出文件中。另外，如何防止添加具有匹配 ID 的行？是否可以替换我的 output.csv 文件中 ID 与 input.csv 文件中的 ID 匹配的行，而不是将具有重复 ID 的行添加到 output.csv？

有人可以帮我解决这个问题吗？提前致谢！

输入文件.csv：

id       date          data1     data2    flag
1     2020-03-01      mydata    mydata1    0
2     2020-03-02      mydata     mydata    1
3     2020-03-03      mydata    mydata1    0

Answer 1

现在我的问题是防止将具有重复 ID 的记录添加到我的 output.csv 中。如果可能的话，我需要用匹配的 ID 覆盖记录。

为了匹配ID，我们无法避免读取输出文件。

import csv

data = dict()
# first read the output file in (if one exists already)
try:
    with open("output file.csv", encoding="utf8") as f_out:
        for row in csv.DictReader(f_out): data[row['id']] = row
except OSError: pass

# now add the new rows from the input file; rows with existing id are replaced
with open("input file.csv", encoding="utf8") as f:
    reader = csv.DictReader(f)
    for row in reader:
        if row['MyColumn'] == '0': data[row['id']] = row

with open("output file.csv", "w", encoding="utf8") as f_out:
    writer = csv.DictWriter(f_out, fieldnames=reader.fieldnames)
    writer.writeheader()
    for row in data: writer.writerow(data[row])

Answer 2

要将新行追加到文件而不是覆盖值，请尝试对文件使用追加 (a) 权限而不是写入 (w)。

with open("/path/to/my/output/file.csv", "a+", encoding="utf8") as f_out:

不需要写t，因为它指的是默认的文本模式。

记录在案的here：

Character   Meaning
    'r'     open for reading (default)
    'w'     open for writing, truncating the file first
    'x'     open for exclusive creation, failing if the file already exists
    'a'     open for writing, appending to the end of the file if it exists
    'b'     binary mode
    't'     text mode (default)
    '+'     open a disk file for updating (reading and writing)
    'U'     universal newlines mode (deprecated)

您问题的第二部分不太清楚。能再详细一点吗？