将列表切分为N等份

def subset(alist, idxs):
    '''
        用法：根据下标idxs取出列表alist的子集
        alist: list
        idxs: list
    '''
    sub_list = []
    for idx in idxs:
        sub_list.append(alist[idx])

    return sub_list
#分层函数
def split_list(alist, group_num=5, shuffle=True, retain_left=False,is_merge = True):
    '''
        用法：将alist切分成group个子列表，每个子列表里面有len(alist)//group个元素
        shuffle: 表示是否要随机切分列表，默认为True
        retain_left: 若将列表alist分成group_num个子列表后还要剩余，是否将剩余的元素单独作为一组
    '''

    index = list(range(len(alist)))  # 保留下标

    # 是否打乱列表
    if shuffle:
        random.shuffle(index)

    elem_num = len(alist) // group_num  # 每一个子列表所含有的元素数量
    sub_lists = {}

    # 取出每一个子列表所包含的元素，存入字典中
    for idx in range(group_num):
        start, end = idx * elem_num, (idx + 1) * elem_num
        sub_lists['set' + str(idx)] = subset(alist, index[start:end])

    # 是否将最后剩余的元素作为单独的一组
    if retain_left and group_num * elem_num != len(index) and is_merge==False:  # 列表元素数量未能整除子列表数,需要将最后那一部分元素单独作为新的列表
        sub_lists['set' + str(idx + 1)] = subset(alist, index[end:])
    #是否将最后剩余的元素合并到最后的一组
    if retain_left and is_merge and group_num * elem_num != len(index):
        sub_lists['set' + str(idx)] += subset(alist, index[end:])
    return sub_lists