使用python或bash将文件排列到文件夹中？

import os
import shutil

def arrange_files(directory):
    # 获取目录下的所有文件
    files = os.listdir(directory)
    
    for file in files:
        # 获取文件的扩展名
        _, ext = os.path.splitext(file)
        
        # 创建对应扩展名的文件夹（如果不存在）
        if not os.path.exists(ext):
            os.makedirs(ext)
        
        # 移动文件到对应的文件夹中
        shutil.move(file, os.path.join(ext, file))

# 调用函数，传入需要排列的文件所在的目录
arrange_files('/path/to/files')

#!/bin/bash

# 指定需要排列的文件所在的目录
directory="/path/to/files"

# 遍历目录下的所有文件
for file in "$directory"/*; do
    # 判断是否为文件
    if [[ -f "$file" ]]; then
        # 获取文件的扩展名
        ext="${file##*.}"
        
        # 创建对应扩展名的文件夹（如果不存在）
        if [[ ! -d "$directory/$ext" ]]; then
            mkdir "$directory/$ext"
        fi
        
        # 移动文件到对应的文件夹中
        mv "$file" "$directory/$ext"
    fi
done