주로 폴더의 파일을 긁어오는 용도로 사용되는데, 경로를 이용해 원하는대로 뽑을 수 있습니다.
form glob import glob
glob('*.exe')
>> ['aaaaaa.exe', 'gggg.exe']
golb('*.txt')
>> ['aaaaaa.txt', 'gggg.txt']
golb()
함수는 인자로 받은 패턴과 이름이 일치하는 모든 파일과 디렉토리의 리스트를 반환합니다.
*
는 전체를 표현합니다. 물론 현재경로가 아닌 다른 경로에 대해서도 조회 할 수 있습니다.
파일을 바꾼다거나 경로를 찾아가는 등 주로 노트북 외부에서 하는 일을 처리하는 모듈입니다.
os.getcwd()
현재경로를 찾을 수 있습니다.
os.listdir()
현재경로의 파일들을 리스트로 뽑아 줍니다.
예시1 )
os.listdir('../dataset/')
>>['aaa', 'aaa.zip', 'asv_png', 'asv_png.zip']
예시2 ) 폴더별 데이터의 개수 비교
label_nums = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
dataset = os.listdir('../dataset/')
nums_dataset = []
for n in label_nums:
data_per_class = os.listdir('../dataset/' + n)
nums_dataset.append(len(data_per_class))
plt.bar(label_nums, nums_dataset)
plt.title('Number of Dataset per class')
plt.show()