import os #this command will store all .txt files in same directories ALL_FILES_IN_DIR = [ELEM for ELEM in os.listdir() if "txt" in ELEM] #ALL DIRETORIES ALL_DIR = [ELEM for ELEM in os.listdir() if "." not in ELEM]