Question

我想编写一个Python脚本，搜索当前目录中的所有文件夹，查找所有.txt个文件，并创建一个文件，该文件是当前目录中所有这些文件（以任何顺序）的串联。如果文件夹包含子文件夹，则不应搜索这些子文件夹。一个例子是

main_folder
  folder_1
    sub_folder
      file1.txt
    file2.txt
  folder_2
    file3.txt

脚本放在main_folder内。它应该在file2.txt内创建一个file3.txt和main_folder（以任何顺序）串联的文件。

我的问题是：如何告诉Python遍历文件夹，查找.txt文件，而无需进入子文件夹？

Answer 1

使用glob：

>>> import glob
>>> glob.glob('main_folder/*/*.txt')
['main_folder/folder_1/file2.txt', 'main_folder/folder_2/file3.txt']

Answer 2

来自the docs：

当 topdown 为True时，调用者可以就地修改 dirnames 列表（可能使用del或切片分配），并且walk()只会递归到名称保留在 dirnames 中的子目录中;这可用于修剪搜索，强制执行特定的访问顺序，甚至可以通知walk()调用者在再次恢复walk()之前创建或重命名的目录。

Answer 3

使用os.walk()功能并将topdown选项设置为True。删除返回的元组中的所有dirnames：

for root, dirs, files in os.walk('/base/dir', topdown=True):
    del dirs[:]
...

Answer 4

如果您使用的是基于Unix的系统..我认为最简单的方法是使用find和cat使用python subprocess模块，就像这样。使用find -depth选项确实有帮助，例如，使用-depth +3查找深度为＆gt; = 3的所有文件

>>> import subprocess
>>> lst = subprocess.check_output('find . -name "*.txt" -depth 2', shell=True)
>>> print lst
./folder_1/f1.txt
./folder_2/f2.txt 
>>> out_f = subprocess.check_output('cat '+lst.replace('\n', ' '), shell=True)
>>> print out_f
inside Folder_1
inside Folder 2

您可以通过单独提供每个参数来避免shell=True。

Python读取目录中的文件并连接

4 个答案: