Python：根据现有列将列添加到 CSV 文件

Question

我已经编写了识别和解析我正在寻找的值所需的内容，我需要帮助将一列写入具有解析值的 csv 文件（或新的 csv 文件）。以下是我正在尝试做的一些伪代码/有点现实的 Python 代码：

# Given a CSV file, this function creates a new CSV file with all values parsed
def handleCSVfile(csvfile):
  with open(csvfile, 'rb') as file:
    reader = csv.reader(file, delimiter=',', lineterminator='\n')
    for row in reader:
        for field in row:
          if isWhatIWant(field):
            parsedValue = parse(field)
            # write new column to row containing parsed value

我已经编写了isWhatIWant 和parse 函数。如果我需要编写一个全新的 csv 文件，那么我不确定如何同时打开并从一个文件读取和写入另一个文件。

Answer 1

我会这样做。我猜isWhatIWant() 是应该就地替换字段的东西。

import csv

def handleCSVfile(infilename, outfilename):
    with open(infilename, 'rb') as infile:
        with open(outfilename, 'wb') as outfile:
            reader = csv.reader(infile, lineterminator='\n')
            writer = csv.writer(outfile, lineterminator='\n')

            for row in reader:
                for field_index, field in enumerate(row):
                    if isWhatIWant(field):
                        row[field_index] = parse(field)
                writer.writerow(row)

这种模式经常出现并导致非常长的行。有时将打开和文件的逻辑分解为不同的函数会很有帮助，如下所示：

import csv

def load_save_csvfile(infilename, outfilename):
    with open(infilename, 'rb') as infile:
        with open(outfilename, 'wb') as outfile:
            reader = csv.reader(infile, lineterminator='\n')
            writer = csv.writer(outfile, lineterminator='\n')

            read_write_csvfile(reader, writer)

def read_write_csvfile(reader, writer)
    for row in reader:
        for field_index, field in enumerate(row):
            if isWhatIWant(field):
                row[field_index] = parse(field)
        writer.writerow(row)

这将代码模块化，使您可以更轻松地从彼此独立的逻辑更改文件和格式的处理方式。

其他提示：

• 不要将变量命名为file，因为这是一个内置函数。当你最不期待的时候，模仿这些名字会咬你。
• delimiter=',' 是默认值，因此您无需明确指定。