如何在python中仅获取cwd的子文件夹

Question

我需要得到我当前工作目录的子文件夹。你知道吗

我尝试了os模块，找到了tree-命令。我只需要得到子文件夹，而不是子文件夹或任何在子文件夹。你知道吗

以下是我目前得到的：

import os

cwd = os.getcwd()
dirctorys  = os.tree(cwd)

我创建了一个for循环来打印我的树，但是它打印所有的子文件夹，包括它的内容。但我只需要子文件夹，没有别的！你知道吗

例如，我的目录：

root/me/cwd
        |-__init__.py
        |-mygeo.py
        |-subfolder1
        |    |-file1.py
        |    |-file2.py
        |-subfolder2
        |    |-file3.py
        |    |-subsubfolder
        |    |    |-foo.py
        |    |    |-foo2.py

我希望python返回一个元组或一个包含以下内容的列表：

['__init__.py', 'mygeo.py', 'subfolder1', 'subfolder2']

我分析了os.tree(cwd)打印时的输出，发现它是由列表组成的。第一份清单似乎正是我需要的，但我不知道如何得到它。你知道吗

我怎么能得到这样的名单？你知道吗

Answer 1

您可以使用os.listdir()：

import os

cwd = os.getcwd()
contents = os.listdir(cwd)

如果要从目录中拆分文件：

import os

cwd = os.getcwd()
_, files, directories = next(os.walk(cwd))