how to get rid of images in python code example

Example: python delete saved image

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