python 文件整理 shutil

以下代碼將指定目錄下的所有文件分類，按照後綴名歸類存儲

import os
import shutil

path = './'
def foo():
    files = os.listdir(path)
    for f in files:
        folder = path + f.split('.')[-1]
        if not os.path.exists(folder):
            os.makedirs(folder)
        shutil.move(f, folder)


if __name__ == '__main__':
    foo()