5

我需要编辑几个 csv 文件。实际上，大多数文件都很好，只是需要更改的最后（第 41 列）列。对于该列中每次出现的特定字符串，我需要将其替换为不同的字符串；具体来说，每次出现的“SD”都需要替换为“S”。我尝试使用 Python 完成此操作，但我认为我需要编写 csv 文件，但我不太确定如何执行此操作：

import os 
import csv


path=os.getcwd()

filenames = os.listdir(path)

for filename in filenames:

    if filename.endswith('.csv'):
        r=csv.reader(open(filename))

        for row in r:
            if row[40] == "S-D":
                row[40] = "S"

有什么帮助吗？此外，如果有人有一种快速、优雅的方式使用 shell 脚本来执行此操作，那可能对我也很有帮助。

4

2 回答 2

4

沿着这些思路尝试一些东西。现在使用glob@SaulloCastro 提到的模块和csv模块。

import glob
import csv

for item in glob.glob(".csv"):
    r = list(csv.reader(open(item, "r")))
    for row in r:
        row[-1] = row[-1].replace("S-D", "S")
    w = csv.writer(open(item, "w"))
    w.writerows(r)
于 2013-10-02T20:12:32.880 回答
3

请务必阅读有关CSV 文件读取和写入的 Python 文档。在那里可以学到很多东西。这是基于您的问题的基本示例。只修改最后一列的数据，写出一个修改后的文件，名字中带有“_edited”。

import os 
import csv


path=os.getcwd()

filenames = os.listdir(path)

for filename in filenames:

    if filename.endswith('.csv'):
        r=csv.reader(open(filename))
        new_data = []
        for row in r:
            row[-1] = row[-1].replace("S-D", "S")
            new_data.append(row)

        newfilename = "".join(filename.split(".csv")) + "_edited.csv"
        with open(newfilename, "w") as f:
            writer = csv.writer(f)
            writer.writerows(new_data)
于 2013-10-02T20:25:11.067 回答