[Python] 특정 폴더의 파일 리스트 가져오기 - Seonmi-Hwang/study GitHub Wiki

다음과 같이 실행하는 .py 파일과 불러오는 폴더는 같은 폴더에 존재한다고 가정한다.

testDigits 폴더의 내부는 다음과 같다.

Student20170995.py

from os import listdir

fileNameList = listdir('testDigits')
m = len(fileNameList)

for i in range(m): # 파일의 개수만큼 반복
  fileNameStr = fileNameList[i]
  answer = int(fileNameStr.split('_')[0]) # 파일 이름의 일부 가져오기
...

Ref https://itholic.github.io/python-listdir-glob/