CODE EXAMPLE FOR PYTHON

List comprehension - list files with extension in a directory

a = [name for name in os.listdir(".") if name.endswith(".txt")]