編程速記(8)：Python篇-獲取文件夾下指定文件後綴名的文件名

basic:os.listdir() & str.endswith()

import os
def get_list_img(data_dir):
	files = os.listdir(data_dir)
	basic_path = data_dir
	for i in files:
		if i.endswith(".png"):
		print(i)