python script to delete files with extension code example

Example 1: python delete file with extension

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

Example 2: python code to remove file extension

import os
print os.path.splitext("sample.txt")[0]