get list of .py files in directory code example

Example 1: python list files in current directory

import os

files = os.listdir('.')
print(files)
for file in files:
  # do something

Example 2: get list of folders in directory python

import os 
my_list = os.listdir('My_directory')