Python分割大量图片到当前文件夹下

主要功能：
      将大量图片等数量的平均分配到不同的文件夹下，如果不能平均分多余的部分保留在当前文件夹下。


代码:
# 主要功能：对于当前文件夹下的图片平均分成几份，分别放到不同的文件夹下
import os
import pandas as pd
import shutil
# 定义函数
# 定义一个读取文件夹下文件名字的函数
def  readname(path):
    filename = os.listdir(path)
    # print(filename)
    return filename


# 定义一个列表分割函数
def list_of_group(photo_len, people, name):
    """
    :param photo_len: 照片数量
    :param people: 分配给几位同学
    如果刚好整除，就平均分给每位同学，如果多出来，直接单独拿出来成一组，放到最后
    :return:
    """
    list_group_mod = photo_len % people
    list_group = (photo_len-photo_len % people)/people
    print("每组分：", int(list_group), "余下来：", int(list_group_mod))

    step = int(list_group)
    for i in range(0, photo_len, step):
        name.append(photo_list[i:i + step])
    return name


# 在当前目录下创建一个文件夹
def create_file(save_path):
    import os
    if os.path.exists(save_path):
        print('该文件夹已存在')
    else:
        print('该文件夹不存在，可以创建')
        os.mkdir(save_path)
        print('{}文件夹创建成功'.format(save_path))
    return


# 开始干活
# 选择好图片路径
# 图片保存的当前路径
file_path = '/Users/~/remove_photo/remove_photo'

file_name = readname(file_path)

# 因为复制文件的时候会产生.DS_Store，所以要将他删除（不删好像也没啥大事儿）QAQ
if '.DS_Store' in file_name:
    print("在文件目录下有 .DS_Store！！！")
    file_name.remove('.DS_Store')
    if '.DS_Store' not in file_name:
        print('.DS_Store删除成功！！！')
else:
    print('.DS_Store删除成功！！！')
try:
    path = file_path + "/" + file_name[0]
    photo_list = readname(path)
    photo_len = int(len(photo_list))

    print("当前文件下总计:", photo_len, "张图片")

    people = int(input("请输入要分给几位同学："))
    # 定义一个空的列表，将图片文件名称保存在列表中
    name = []
    list_of_group(photo_len, people, name)
    # 创建文件夹
    for i in range(people):
        create_file(file_path + "/" + str(i+1))
        # 将分隔好的数组插入到指定的文件夹中
        for j in name[i]:
            # shutil.move(source, destination)
            # source: 代表源文件路径的字符串。
            # destination: 代表目标目录路径的字符串。
            # 将图片移动到新生成的文件夹中
            shutil.move(path + '/' + j, file_path + '/' + str(i+1))
            print(file_path + '/' + str(i+1))
    print("程序运行成功！！！")
except():
    print('程序运行错误')