ลบแฟ้มกับส่วนเสริมเดียวกัน

ลบแฟ้มกับส่วนเสริมเดียวกัน

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