我有数百个没有标头的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_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()
是处理标头数据和更改列名的方法,我只提供一种设置新索引名的可能方法。注意:
我确实建议您将修改后的文件保存在新文件名下的新文件夹中,因此,如果出现问题,请始终使用原始文件进行备份。
关于python - 如何将列标题批量添加到目录中的所有CSV文件中,并保留这些文件?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/57452378/