### 获取当前目录下所有文本文件的方法 在 Python 中，可以使用多种方法来获取与脚本位于同一目录下的所有文本文件。以下是一个完整的解决方案： #### 使用 `os` 模块和 `glob` 模块 通过结合 `os` 和 `glob` 模块，可以轻松实现这一功能。以下代码展示了如何获取当前目录下的所有 `.txt` 文件[^4]。 ```python import os import glob # 获取当前脚本所在的目录 current_directory = os.path.dirname(os.path.abspath(__file__)) # 使用 glob 模块匹配当前目录下的所有 .txt 文件 text_files = glob.glob(os.path.join(current_directory, "*.txt")) # 打印所有文本文件的路径 for file in text_files: print(file) ``` 上述代码中，`os.path.dirname(os.path.abspath(__file__))` 用于获取当前脚本所在的目录[^1]。`glob.glob` 函数则用于匹配指定模式的文件名，此处匹配的是以 `.txt` 结尾的文件[^4]。 #### 使用 `os.walk` 遍历当前目录 如果需要更灵活的方式，可以使用 `os.walk` 来遍历当前目录及其子目录中的所有文本文件[^3]。 ```python import os # 获取当前脚本所在的目录 current_directory = os.path.dirname(os.path.abspath(__file__)) # 使用 os.walk 遍历当前目录及其子目录 text_files = [] for root, dirs, files in os.walk(current_directory): for file in files: if file.endswith(".txt"): # 判断是否为 .txt 文件 text_files.append(os.path.join(root, file)) # 打印所有文本文件的路径 for file in text_files: print(file) ``` 这段代码中，`os.walk` 返回一个三元组 `(root, dirs, files)`，其中 `root` 是当前遍历的目录路径，`dirs` 是该目录下的子目录列表，`files` 是该目录下的文件列表。通过检查文件名是否以 `.txt` 结尾，可以筛选出所有文本文件[^3]。 #### 使用 `mimetypes` 模块判断文件类型 如果需要更精确地判断文件是否为文本文件，可以结合 `mimetypes` 模块中的 `guess_type` 函数[^5]。 ```python import os import mimetypes # 获取当前脚本所在的目录 current_directory = os.path.dirname(os.path.abspath(__file__)) # 遍历当前目录下的所有文件 all_files = [os.path.join(current_directory, f) for f in os.listdir(current_directory) if os.path.isfile(os.path.join(current_directory, f))] # 筛选出文本文件 text_files = [f for f in all_files if mimetypes.guess_type(f)[0] == 'text/plain'] # 打印所有文本文件的路径 for file in text_files: print(file) ``` 此代码片段中，`mimetypes.guess_type(f)[0]` 返回文件的 MIME 类型，如果是 `text/plain`，则说明该文件是文本文件。 ### 注意事项 - 如果脚本运行在交互式环境中（如 Jupyter Notebook 或 REPL），`__file__` 可能不可用，此时可以使用 `os.getcwd()` 获取当前工作目录。 - 在某些情况下，可能需要处理隐藏文件或符号链接，需根据具体需求调整代码逻辑。