Question

我需要编辑几个csv文件。实际上，大多数文件都是正常的，它只是需要更改的最后一个（第41个）列。对于该列中特定字符串的每次出现，我需要将其替换为不同的字符串;具体而言，每次出现'S-D'都需要用'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脚本快速，优雅地做到这一点，那对我来说也许会非常有帮助。

Answer 1

请务必阅读CSV File Reading and Writing的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)

Answer 2

沿着这些方向尝试一些事情。现在使用@SaulloCastro和glob模块提到的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)

使用Python批量编辑csv文件

2 个答案: