python 中将大列表拆分成小列表

一、将大列表拆分为小列表

def split_list(li, n):
    """
    将列表拆分成每n个元素为一个小列表的[[],[],[]]格式
    args:
        li: 需要分割的列表
        n: 需要分割后每个小列表中的元素数量
    """
    # 分割点的下标
    index_li = []
    for d in li[::n]:
        index = li.index(d)
        index_li.append(index)
    # 每个小列表
    item_li = []
    for i in index_li:
        l = []
        if i != index_li[-1]:
            for item in li[i:i+n]:
                l.append(item)
        else:
            for item in li[i:len(li)]:
                l.append(item)
        item_li.append(l)
    return item_li

猜你喜欢

转载自www.cnblogs.com/Frange/p/9140004.html