Python -导入和重写多个CSV文件

xwmevbvl  于 2023-02-27  发布在  Python
关注(0)|答案(1)|浏览(147)

我有一些csv文件,我想读,删除空列和重写每个到一个单独的文件。
到目前为止,在Jupyter Notebook中,我可以为单个文件执行此操作,没有任何问题:

from pandas.io.parsers import read_csv
data = read_csv('C:\\Users\\tom\\Desktop\\plugs\\20_80-1_plugs.csv')
filtered_data = data.dropna(axis='columns', how='all')
filtered_data.to_csv('C:\\Users\\tom\\Desktop\\plugs\\20_80-1_plugs_modify.csv'), index=False)

有没有人可以帮助扩展这个函数来读取文件夹中的下一个csv文件,并使用与输入相同的名称写入(比如在文件名中使用_modify)。文件夹将包含以下文件:
20_80- 1_插件. csv
11_91- 9_插件. csv
55_21- 2_插件. csv
等等

mepcadol

mepcadol1#

如果要编辑文件夹中的所有文件,可以执行以下操作

import os
import pandas as pd

root_folder = 'C:\\Users\\tom\\Desktop\\plugs\\'

for file in os.listdir(root_folder):
    data = pd.read_csv(os.path.join(root_folder, file))
    filtered_data = data.dropna(axis='columns', how='all')
    new_name = file.replace('.csv', '_modified.csv')
    filtered_data.to_csv(os.path.join(root_folder, new_name), index=False)

基本上,您所做的是遍历目录中的所有文件,复制数据,执行所需的操作,然后将数据保存到相同的位置,但将.csv替换为_modified.csv(添加所需的修改标记)

相关问题