Python | Arrange the files in directories according to extensions

In this article, we will learn how to arrange the files in different directories according to their extensions. When you have a lot of files with different extension than you can simply use the script for your help.

Below is the Python implementation –

import os  # Using system functions

import shutil # Moving file to different folders.

# Changing the folder location.
os.chdir(os.getcwd())
cwd = os.getcwd()

# For listing files from the folder.

l = [f for f in os.listdir(cwd) if os.path.isfile(f)]
l2 = []

# Get the extension of file from list l.

for  value in l:
    s = value.split('.')[1]
    l2.append(s)
print(l, l2)

# We remove duplicate values from
# list l2 and checks if directory
# exists otherwise we create new one

for extension in set(l2):
    dirname = extension
    if os.path.exists(cwd+'\'+extension):
        pass
    else:
        os.makedirs(dirname)

# We use zip function and list l and
# l2 passed as arguments.
# If extension in file is same and
# file not exists then we move the file.

for files, extension in zip(l, l2):
    if extension in files:
        if os.path.exists(cwd+'\'+extension+'\'+files):
            pass
        else:
            shutil.move(cwd+'\'+files, cwd+'\'+extension)
        print(extension, files)
    else :
        print('error')

Files Not Arranged:

Files Arranged:

Output:

['arrange.py', 'mydata.pdf', 'python.exe', 'resume.doc'] ['py', 'pdf', 'exe', 'doc']
py arrange.py
pdf mydata.pdf
exe python.exe
doc resume.doc