[python] 현재 디렉토리의 모든 서브 디렉토리 목록 가져 오기

여러 os.path.join ()이 필요없고 전체 경로를 직접 얻을 수 있기 때문에 위의 것보다 훨씬 좋습니다. 원하는 경우 Python 3.5 이상 에서이 작업을 수행 할 수 있습니다 .

subfolders = [ f.path for f in os.scandir(folder) if f.is_dir() ]

하위 디렉토리의 전체 경로를 제공합니다. 하위 디렉토리의 이름 만 원한다면 f.name대신f.path

https://docs.python.org/3/library/os.html#os.scandir

약간 OT : 모든 하위 폴더를 재귀 적으로 필요로 하는 경우 및 / 또는 모든 파일을 재귀 적 으로 필요로하는 경우이 기능을 살펴보십시오.이 기능은 os.walk& 보다 빠르며 glob모든 하위 폴더의 목록뿐만 아니라 (하위) 하위 폴더 내의 모든 파일을 반환합니다. https://stackoverflow.com/a/59803793/2441026

모든 하위 폴더 만 재귀 적으로 원할 경우 :

def fast_scandir(dirname):
    subfolders= [f.path for f in os.scandir(dirname) if f.is_dir()]
    for dirname in list(subfolders):
        subfolders.extend(fast_scandir(dirname))
    return subfolders

전체 경로가있는 모든 하위 폴더의 목록을 반환합니다. 이것은 다시보다 빠릅니다os.walk 훨씬 빠릅니다 glob.

모든 기능의 분석

tl; dr :
– 폴더에 대한 모든 직접적인 하위 디렉토리 를 얻으려면을 사용하십시오 os.scandir.
– 당신이 얻고 싶은 경우에 모든 하위 디렉토리, 심지어 중첩 된 것들, 사용 os.walk약간 빠른 – – 또는 ▶ fast_scandir위의 함수를.
– os.walk수백 (!) 배보다 느릴 수 있으므로 최상위 서브 디렉토리에만 사용하지 마십시오 os.scandir.

• 아래 코드를 실행하면 OS가 폴더에 액세스하고 결과를 버리고 테스트를 실행하도록 코드를 한 번 실행해야합니다. 그렇지 않으면 결과가 고정됩니다.
• 함수 호출을 혼합하고 싶을 수도 있지만 테스트했지만 실제로는 중요하지 않았습니다.
• 모든 예제는 폴더의 전체 경로를 제공합니다. (Windows) Path 객체 인 pathlib 예제입니다.
• 첫 번째 요소 os.walk는 기본 폴더입니다. 따라서 하위 디렉토리 만 얻을 수는 없습니다. 당신이 사용할 수있는fu.pop(0)제거하는 데 .
• 어떤 결과도 자연 정렬을 사용하지 않습니다 . 즉, 결과는 1, 10, 2로 정렬됩니다. 자연 정렬 (1, 2, 10) 을 얻으려면 https://stackoverflow.com/a/48030307/2441026 을 참조하십시오.

결과 :

os.scandir      took   1 ms. Found dirs: 439
os.walk         took 463 ms. Found dirs: 441 -> it found the nested one + base folder.
glob.glob       took  20 ms. Found dirs: 439
pathlib.iterdir took  18 ms. Found dirs: 439
os.listdir      took  18 ms. Found dirs: 439

W7x64, Python 3.8.1로 테스트되었습니다.

# -*- coding: utf-8 -*-
# Python 3


import time
import os
from glob import glob
from pathlib import Path


directory = r"<insert_folder>"
RUNS = 1


def run_os_walk():
    a = time.time_ns()
    for i in range(RUNS):
        fu = [x[0] for x in os.walk(directory)]
    print(f"os.walk\t\t\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms. Found dirs: {len(fu)}")


def run_glob():
    a = time.time_ns()
    for i in range(RUNS):
        fu = glob(directory + "/*/")
    print(f"glob.glob\t\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms. Found dirs: {len(fu)}")


def run_pathlib_iterdir():
    a = time.time_ns()
    for i in range(RUNS):
        dirname = Path(directory)
        fu = [f for f in dirname.iterdir() if f.is_dir()]
    print(f"pathlib.iterdir\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms. Found dirs: {len(fu)}")


def run_os_listdir():
    a = time.time_ns()
    for i in range(RUNS):
        dirname = Path(directory)
        fu = [os.path.join(directory, o) for o in os.listdir(directory) if os.path.isdir(os.path.join(directory, o))]
    print(f"os.listdir\t\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms. Found dirs: {len(fu)}")


def run_os_scandir():
    a = time.time_ns()
    for i in range(RUNS):
        fu = [f.path for f in os.scandir(directory) if f.is_dir()]
    print(f"os.scandir\t\ttook {(time.time_ns() - a) / 1000 / 1000 / RUNS:.0f} ms.\tFound dirs: {len(fu)}")


if __name__ == '__main__':
    run_os_scandir()
    run_os_walk()
    run_glob()
    run_pathlib_iterdir()
    run_os_listdir()

답변

d='.'
filter(lambda x: os.path.isdir(os.path.join(d, x)), os.listdir(d))

import os

print("root prints out directories only from what you specified")
print("dirs prints out sub-directories from root")
print("files prints out all files from root and directories")
print("*" * 20)

for root, dirs, files in os.walk("/var/log"):
    print(root)
    print(dirs)
    print(files)