현재 디렉터리에서 파일 목록 출력하기

현재 디렉토리에서 특정 확장자의 파일 목록을 가져오는 코드

[filelist.py]

#-*- coding:utf-8 -*-
import os.path

folder=os.getcwd()
print 'current directory :%s' % folder

for filename in os.listdir(folder):
	ext=filename.split('.')[-1]
	if ext == 'exe':
		print filename

[출력결과]

current directory :C:\Users\K\Desktop\malware
Lab01-01.exe
Lab01-02.exe
Lab01-03.exe
Lab01-04.exe