How to delete a file by extension in Python?

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))

For this operation you need to append the file name on to the file path so the command knows what folder you are looking into.

You can do this correctly and in a portable way in python using the os.path.join command.
For example:

import os

directory = "/Users/ben/downloads/"
test = os.listdir( directory )

for item in test:
    if item.endswith(".zip"):
        os.remove( os.path.join( directory, item ) )