os.listdir - 파이썬 디렉토리 내 파일 확인, 개수 새기, 파일 다루기

파이썬 내 os 라이브러리를 이용해 폴더 내의 파일명을 나열할 수 있다.

• os.listdir('폴더 경로')

ex)

import os

file_list = os.listdir('/home/haeyoung/nia12/nia_dataset/data_preprocess/fr_data/')

wavfile = []
txtfile = []
g2pfile = []
etc = []

for file in file_list :
    if file.startswith('g2p') and file.endswith('.txt'):
        g2pfile.append(file)
    elif file.endswith('.wav'):
        wavfile.append(file)
    elif file.endswith('.txt'):
        txtfile.append(file)
    else :
        etc.append(file)
    
print(len(txtfile))
print(len(wavfile))
print(len(g2pfile))
print(len(etc))

 

폴더 내의 파일 명이 리스트 형식으로 담긴다.