循环读取文件夹python

原创

mob649e815a6b81 2023-10-01 06:19:58 ©著作权

文章标签 读取文件数据数据处理 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e815a6b81的原创作品，请联系作者获取转载授权，否则将追究法律责任

循环读取文件夹中的文件并处理数据

引言

在日常的编程工作中，我们经常需要处理大量的数据文件。有时候，这些文件可能存放在同一个文件夹中，我们需要对文件夹中的每个文件进行处理。本文将介绍如何使用Python编程语言来循环读取文件夹中的文件，并进行一些简单的数据处理。

准备工作

在开始编写代码之前，我们需要准备一个包含数据文件的文件夹。假设我们的数据文件存放在名为data的文件夹中，文件夹下有多个以.csv为扩展名的文件。我们将使用Python的os和csv模块来读取和处理这些数据文件。

代码示例

首先，我们需要导入所需的模块：

import os
import csv

之后，我们可以定义一个函数来循环读取文件夹中的文件并处理数据。以下是一个示例函数：

def process_files(folder_path):
    # 获取文件夹中的所有文件
    file_list = os.listdir(folder_path)
    
    # 循环遍历文件夹中的每个文件
    for file_name in file_list:
        # 检查文件扩展名是否为.csv
        if file_name.endswith('.csv'):
            # 构建文件的完整路径
            file_path = os.path.join(folder_path, file_name)
            
            # 打开文件
            with open(file_path, 'r') as file:
                # 使用csv模块读取文件中的数据
                csv_reader = csv.reader(file)
                
                # 循环遍历文件中的每一行数据
                for row in csv_reader:
                    # 在此处进行数据处理操作，例如打印每一行数据
                    print(row)

在上述代码中，process_files函数接受一个文件夹路径作为参数，并使用os.listdir函数获取文件夹中的所有文件。然后，我们使用for循环遍历文件夹中的每个文件，并使用if语句检查文件扩展名是否为.csv。如果是，我们构建完整的文件路径，并使用csv模块的reader函数打开文件。最后，我们使用第二个for循环遍历文件中的每一行数据，并在此处进行我们需要的数据处理操作。

应用实例

假设我们有一个名为data的文件夹，其中存放着几个.csv文件。这些文件中的每一行代表一个学生的信息，包括姓名、年龄和成绩。我们可以使用上述代码来循环读取文件夹中的文件，并处理每个学生的信息。

下面是一个完整的示例代码：

import os
import csv

def process_files(folder_path):
    file_list = os.listdir(folder_path)
    for file_name in file_list:
        if file_name.endswith('.csv'):
            file_path = os.path.join(folder_path, file_name)
            with open(file_path, 'r') as file:
                csv_reader = csv.reader(file)
                for row in csv_reader:
                    print(row)

# 指定文件夹路径
folder_path = 'data'

# 调用函数处理文件夹中的文件
process_files(folder_path)

在这个示例中，我们指定了文件夹路径为data，然后调用process_files函数来处理文件夹中的所有.csv文件。对于每个文件，我们只是简单地打印出每一行的数据。你可以根据自己的需求修改代码来进行更复杂的数据处理操作。