Python|根据扩展名排列目录中的文件

在本文中，我们将学习如何根据扩展名排列不同目录中的文件。当您有很多具有不同扩展名的文件时，您可以简单地使用脚本来寻求帮助。

下面是Python的实现——

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

未整理的文件：

整理的文件：

输出：

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

在评论中写代码？请使用 ide.geeksforgeeks.org，生成链接并在此处分享链接。