How to delete a file by extension in Python?

Question

I was messing around just trying to make a script that deletes items by ".zip" extension.

import sys import os from os import listdir  test=os.listdir("/Users/ben/downloads/")  for item in test:     if item.endswith(".zip"):         os.remove(item) 

Whenever I run the script I get:

OSError: [Errno 2] No such file or directory: 'cities1000.zip' 

cities1000.zip is obviously a file in my downloads folder.

What did I do wrong here? Is the issue that os.remove requires the full path to the file? If this is the issue, than how can I do that in this current script without completely rewriting it.

Answer 1

You can set the path in to a dir_name variable, then use os.path.join for your os.remove.

import os  dir_name = "/Users/ben/downloads/" test = os.listdir(dir_name)  for item in test:     if item.endswith(".zip"):         os.remove(os.path.join(dir_name, item))