Python을 사용하여 디렉터리의 파일만 효율적으로 나열하려면 어떻게 해야 합니까?

디렉터리에서 파일 가져오기

디렉터리의 파일 목록을 검색하는 것은 Python 프로그래밍에서 일반적인 작업입니다. 이를 수행하는 한 가지 방법은 파일과 디렉터리를 모두 포함하여 디렉터리 내의 모든 항목을 반환하는 os.listdir()을 사용하는 것입니다.

접근 방법 1: os.path.isfile() 사용

파일만 나열하려면 다음을 사용하세요. os.path.isfile():

from os import listdir
from os.path import isfile, join
mypath = 'path/to/directory'
onlyfiles = [f for f in listdir(mypath) if isfile(join(mypath, f))]

접근 방법 2: os.walk()

os.walk() 사용은 디렉터리와 해당 하위 디렉터리를 반복합니다. , 방문하는 각 디렉토리에 대해 튜플을 반환합니다. 이 튜플에는 파일용 목록, 디렉터리용 목록, 현재 디렉터리 자체용 목록 등 세 가지 목록이 포함되어 있습니다.

from os import walk
mypath = 'path/to/directory'
filenames = next(walk(mypath), (None, None, []))[2]  # [] if no files

이 접근 방식을 사용하면 디렉터리 구조를 더욱 완벽하게 표현할 수 있습니다. 현재 디렉터리의 파일만 나열하려면 첫 번째 반복 후 루프를 중단하세요.

from os import walk

f = []
for (dirpath, dirnames, filenames) in walk(mypath):
    f.extend(filenames)
    break

위 내용은 Python을 사용하여 디렉터리의 파일만 효율적으로 나열하려면 어떻게 해야 합니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!