python - 如何向目录中的所有 CSV 文件批量添加列标题并保留这些文件？

Question

我有数百个没有标题的 CSV 文件，这些文件是从我使用的某些软件中导出的。列数和确切的列标题可能因批处理而异，但在批处理内绝不会发生变化。

我正在学习 Pandas，我需要一些帮助来整理一个非常简单的笔记本，该笔记本加载目录中的所有 CSV 文件，并将我选择的列标题添加到该目录中的所有文件中，并将它们保存为相同的 CSV文件(相同名称)，但现在文件中包含 header 。

正如我所说，某些批处理在需要标题的列数以及标题内容方面会有所不同，因此保留随意更改标题的能力会很好。

我有以下代码，它适用于一个文件。如何循环遍历目录中的所有文件，添加相同的 header ，然后保存文件。

import pandas as pd
df_csv = pd.read_csv('/Users/F/Desktop/FPython/File1.csv', names=['A', 'B', 'C'])
df_csv.to_csv('/Users/F/Desktop/FPython/File1.csv', index=False)

Answer 1

尝试使用pandas模块，特别是read_csv和 to_csv方法。这样您就可以使用所需的标题作为列名称来修改导入的数据框，然后将修改后的数据框保存回 csv。

您可以使用 glob 模块迭代文件夹中的所有 .csv 文件:

import glob
import pandas as pd

files = glob.glob('./*.csv')

def manipulate_headers(df):
    df.set_axis(['A', 'B', 'C', 'D', 'E', 'F'], axis=1, inplace=True)
    return df

for file_name in files:
    df = pd.read_csv(file_name)
    df = manipulate_headers(df)
    df.to_csv(file_name)

其中 manipulate_headers() 是处理标题数据和更改列名称的方法，我只是提供一种设置新索引名称的可能操作。

注意:
我确实建议您将修改后的文件以新文件名保存在新文件夹中，这样您就可以始终保留原始文件的备份，以防出现问题。