我有数百个没有标头的CSV文件,这些标头从我使用的某些软件中导出。列的数量和确切的列标题在批次之间可能会有所不同,但在批次内绝不会发生变化。

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

就像我说的,某些批次的需要标题的列数和标题将有所不同,因此保留随意更改标题的功能将是很好的。

我有以下代码,它与一个文件一起使用效果很好。如何遍历目录中的所有文件,添加相同的标题,然后保存文件。

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)

最佳答案

尝试使用pandas模块,特别是read_csvto_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()是处理标头数据和更改列名的方法,我只提供一种设置新索引名的可能方法。

注意:
我确实建议您将修改后的文件保存在新文件名下的新文件夹中,因此,如果出现问题,请始终使用原始文件进行备份。

关于python - 如何将列标题批量添加到目录中的所有CSV文件中,并保留这些文件?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/57452378/

10-15 00:19
查看更多