时间戳

首页 > 解决方案 > 在python中读取多个文件夹类别中的多个.txt文件的方法

python - 在python中读取多个文件夹类别中的多个.txt文件的方法

问题描述

我是 Python 新手，正在尝试读取存储在多个文件夹层次结构中的 .txt 文件数据集。文件夹的结构是

-Folder1 
   -Category1_Folder
        -file1.txt
   -Category2_Folder
        -file1.txt
        -file2.txt and so on...

这些类别具有重要意义。我需要能够识别哪个文件属于哪个类别。然后我需要删除停用词并使用 TfIDf 执行特征提取。做这样的事情最简单的方法是什么？

标签： pythonfiledirectory

解决方案

我推荐os.walk。

如果你有这样的目录：

project/
- folder1/
  - file1.png
  - file2.jpg
- folder2/
  - file3.zip

然后，示例代码是：

import os

for dirpath, dirnames, filenames in os.walk(os.getcwd()):  # getcwd() for current work dir
  print(dirpath, dirnames, filenames)

输出来：

/project ['folder1', 'folder2'] []
/project/folder1 [] ['file1.png', 'file2.jpg']
/project/folder2 [] ['file3.zip']

如果您需要文件夹、文件名，请使用 for 循环：

for dirname in dirnames:
  for filename in filenames:
    # split dirname for categories
    # and so on..

推荐阅读