delete image from directory in python code example

Example 1: python delete file

import os
import shutil

if os.path.exists("demofile.txt"):
  os.remove("demofile.txt") # one file at a time

os.rmdir("test_directory") # removes empty directory
shutil.rmtree("test_directory") # removes not empty directory and its content

Example 2: python delete saved image

import os
os.remove(file) for file in os.listdir('path/to/directory') if file.endswith('.png')