python如何遍历两次文件

原创

mob64ca12e33720 2023-12-15 11:25:28 ©著作权

文章标签 python 遍历文件夹文件名 文章分类 Python 后端开发

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

Python如何遍历两次文件

在Python中，我们经常需要遍历文件夹或文件来处理数据。有时候，我们需要对两个文件进行比较或合并操作。本文将介绍如何遍历两次文件，以解决一个具体的问题。

问题描述

假设我们有两个文件夹，分别为folder1和folder2，每个文件夹中包含若干个文件。我们需要比较这两个文件夹中的文件，找出它们的交集和差集。

解决方案

要解决这个问题，我们可以使用os模块来遍历文件夹中的文件。具体步骤如下：

1. 导入模块

首先，我们需要导入os模块来操作文件和文件夹。

import os

2. 定义文件夹路径

我们需要定义两个文件夹的路径，即folder1和folder2。

folder1 = 'path/to/folder1'
folder2 = 'path/to/folder2'

3. 遍历文件夹中的文件

使用os.listdir函数可以获取文件夹中的所有文件名。我们可以通过遍历这些文件名来处理文件。

# 定义一个空列表用于存储文件名
file_names = []

# 遍历folder1中的文件
for file_name in os.listdir(folder1):
    # 获取文件的完整路径
    file_path = os.path.join(folder1, file_name)
    # 将文件的完整路径添加到列表中
    file_names.append(file_path)

# 遍历folder2中的文件
for file_name in os.listdir(folder2):
    # 获取文件的完整路径
    file_path = os.path.join(folder2, file_name)
    # 将文件的完整路径添加到列表中
    file_names.append(file_path)

4. 处理文件

现在，我们可以遍历file_names列表，对每个文件进行处理。在这个例子中，我们将打印交集和差集。

# 定义一个空集合用于存储交集
intersection = set()

# 定义一个空集合用于存储差集
difference = set()

# 遍历file_names列表
for file_path in file_names:
    # 通过文件名来判断文件属于哪个文件夹
    if folder1 in file_path and folder2 in file_path:
        # 如果文件同时存在于两个文件夹中，则将其添加到交集中
        intersection.add(file_path)
    elif folder1 in file_path:
        # 如果文件仅存在于folder1中，则将其添加到差集中
        difference.add(file_path)
    elif folder2 in file_path:
        # 如果文件仅存在于folder2中，则将其添加到差集中
        difference.add(file_path)

# 打印交集
print('交集:')
for file_path in intersection:
    print(file_path)

# 打印差集
print('差集:')
for file_path in difference:
    print(file_path)