Simple gist for opening directory, sorting and writing text

sort_create_label_write_text.py

import glob
from PIL import Image

img_dir_path = 'data_cheek/*.jpg'
train_test_filelist = sorted(glob.glob(img_dir_path))
cheek_label_list = [0 if "old" in files else 1 for files in train_test_filelist]

thefile = open('data_cheek.txt', 'w')

for item in cheek_label_list:
    thefile.write("%s\n" % item)