Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python에서 확장자가 .txt인 디렉토리의 모든 파일을 찾는 방법은 무엇입니까?

<시간/>

os.listdir 메소드를 사용하여 디렉토리의 모든 디렉토리와 파일을 가져올 수 있습니다. 그런 다음 목록을 필터링하여 파일만 가져오고 확장자도 확인합니다.

예를 들어

>>> import os
>>> file_list = [f for f in os.listdir('.') if os.path.isfile(os.path.join('.', f)) and f.endswith('.txt')]
>>> print file_list
['LICENSE.txt', 'NEWS.txt', 'README.txt']

endwith 메소드는 문자열이 특정 접미사로 끝나는지 확인하는 문자열 클래스의 멤버입니다.

glob 모듈을 사용하여 동일한 결과를 얻을 수도 있습니다.

>>> import glob, os
>>> file_list = [f for f in glob.glob("*.txt")]
>>> print file_list
['LICENSE.txt', 'NEWS.txt', 'README.txt']