python - 将子文件夹的名称保留在列表中

Question

在这个变量中有一个名为main_folder. 此文件夹有两个文件夹：

111,222

我需要在列表中获取这些文件夹的名称。

试过这个：

a = r'C:\Users\user\Desktop\main_folder'
import os 
for root, dirs, files in os.walk(a):
    print(dirs)

给出：

['111', '222'] # <--------------This only needed
[]
[]

如何只保留第一个列表而不是我认为描述这些文件夹内容的空列表，因为它们没有文件夹。

标签： python

Answer 1

通过以下方式迭代一次next：

import os

a = r'C:\Users\user\Desktop\main_folder'
walker = os.walk(a)
res = next((dirs for _, dirs, _ in walker), [])

walker如有必要，您可以通过其他next调用继续迭代。