python iterate through directories in a directory code example

Example 1: python iterate directory

import os

directory = r'C:\Users\admin'
for filename in os.listdir(directory):
    if filename.endswith(".jpg") or filename.endswith(".png"):
        print(os.path.join(directory, filename))
    else:
        continue

Example 2: traverse files in directory python

# python 3
import os

for dirpath, dirnames, filenames in os.walk(dir_path):
    print(dirpath)
    print(dirnames)
    print(filenames)
    
# python 2
import os.path

def print_it(x, dir_name, files):
    print(dir_name)
    print(files)

os.path.walk(your_dir, print_it, 0)