to get only needed files from the folder function in python code example

Example 1: get files in directory python

import os
files_and_directories = os.listdir("path/to/directory")

Example 2: list directory in python

from os import listdir

## Prints the current directory as a list (including file types)
print(os.listdir())