获取以Python结尾的列表文件

当我们需要操作一些特定类型的文件时，通常需要先筛选出符合条件的文件。本文介绍如何使用Python筛选出以.py结尾的文件，将它们添加到列表中并返回该列表。

代码示例

import os

def get_py_files(directory):
    '''
    获取以py结尾的文件的列表
    :param directory: 需要查找文件的目录
    :return: 符合条件的文件列表
    '''
    py_files = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith('.py'):
                py_files.append(os.path.join(root, file))
    return py_files

代码说明

以上代码使用了Python的os模块，通过调用os.walk()遍历了给定目录下的所有文件和文件夹。对于每一个文件，我们检查它的文件名是否以.py结尾。如果是，就把它的路径添加到列表中。

当函数完成遍历并获取所有符合条件的文件路径后，将它们以列表的形式返回。

使用示例

directory = '/path/to/directory' # 指定目录
py_files = get_py_files(directory) # 获取符合条件的文件列表
print(py_files) # 打印列表

结论

使用以上代码，我们可以简单方便地获取以.py结尾的文件并将它们加入列表，以备之后的操作。