Python - 修改 csv 文件

Question

现在我知道在读取 csv 文件时修改它通常是不可行的，因此您需要创建一个新的 csv 文件并写入它。我遇到的问题是保留数据的原始顺序。

输入的 csv 文件如下所示：

C1       C2         C3
apple    BANANA     Mango
pear     PineApple  StRaWbeRRy

我想将所有数据转成小写并输出一个新的 csv 文件，如下所示：

C1       C2         C3
apple    banana     mango
pear     pineapple  strawberry

到目前为止，我可以遍历输入 csv 文件并将所有值转换为小写，但我不知道如何将其重写为该格式的 csv 文件。我的代码是：

def clean (input)
  aList = []
  file = open(input, "r")
  reader = csv.reader(file, delimiter = ',')
  next(reader, None) # Skip the header but I want to preserve it in the output csv file
  for row in reader:
     for col in row:
        aList.append(col.lower())

所以现在我有一个包含所有小写数据的列表，如何将其重写为与输入格式相同（行数和列数相同）的 csv 文件，包括我在代码中跳过的标题行.

Answer 1

如果您只想更改数据的大小写并保留其他所有内容，您最好跳过 csv 模块并使用 直接文件 例如：

# Open both files
with open("infile.csv") as f_in, open("outfile.csv", 'w') as f_out:
    # Write header unchanged
    header = f_in.readline()
    f_out.write(header)

    # Transform the rest of the lines
    for line in f_in:
        f_out.write(line.lower())

Answer 2

熊猫方式：

使用pandas 读取文件并获取数据框。然后你可以简单地使用lower()

import pandas as pd

def conversion(text):
    return text.lower()
    

df = pd.read_csv(file_path)
df[column_name] = df[column_name].map(conversion)

甚至是一个班轮：

df[column_name] = df[column_name].apply(lambda x: x.lower()) # If you have nan or other non-string values, you may need to convert x to string first like str(x).lower()

然后你可以使用to_csv函数保存它

Answer 3

如果你想全部使用 csv 模块，请使用以下代码 sn-p。

import os
import csv


def clean(input):
    tmpFile = "tmp.csv"
    with open(input, "r") as file, open(tmpFile, "w") as outFile:
        reader = csv.reader(file, delimiter=',')
        writer = csv.writer(outFile, delimiter=',')
        header = next(reader)
        writer.writerow(header)
        for row in reader:
            colValues = []
            for col in row:
                colValues.append(col.lower())
            writer.writerow(colValues)
    os.rename(tmpFile, input)

Answer 4

我发现的最简单的方法如下 让最初的CSV文件名是test.csv

with open('test.csv','r') as f:
    with open('cleaned.csv','w') as ff:
        ff.write(f.readline())
        ff.write(f.read().lower())

以上代码将创建一个全小写的新 csv