列表分割

柿子饼同学

就是有一个列表,像这样:

l = [1, 2, 3, 1, 2, 2, 3, 1, 5, 6, 3, 1, 3, 1]

现在想做的就是把他分割成由1打头的小列表,像这样

[[1, 2, 3], [1, 2, 2, 3], [1, 5, 6, 3], [1, 3], [1]]

这该怎么办

kogawananari

用高阶函数reduce

不是黄瓜是傻瓜

如图

qq1151985918

l = [1, 2, 3, 1, 2, 2, 3, 1, 5, 6, 3, 1, 3, 1]

def f(lst):
    res = [[] for i in range(lst.count(1))]
    flg = -1
    for i in lst:
        if i == 1:
            flg += 1
        if flg == -1:
            res.append(i)
            continue
        res[flg].append(i)
    return res

print(f(l))

qq1151985918

l = [1, 2, 3, 1, 2, 2, 3, 1, 5, 6, 3, 1, 3, 1]

def f(lst):
    res = [[] for i in range(lst.count(1))]
    flg = -1
    for i in lst:
        if i == 1:
            flg += 1
        if flg == -1:
            res.append(i)
            continue
        res[flg].append(i)
    return res

print(f(l))

qq1151985918

应该是在审核，不知道为啥

柿子饼同学

qq1151985918 发表于 2021-4-11 19:55
应该是在审核，不知道为啥

没看懂

柿子饼同学

kogawananari 发表于 2021-4-11 19:46
用高阶函数reduce

能详细说说么

kogawananari

柿子饼同学 发表于 2021-4-11 20:01
能详细说说么

from functools import reduce

l = [1, 2, 3, 1, 2, 2, 3, 1, 5, 6, 3, 1, 3, 1]

def ss1(li):
    def set_1(a, v):
        a[-1].append(v) if v!=1 else a.append([v])
        return a

    return list(reduce(set_1, li, []))

print(ss1(l)) #[[1, 2, 3], [1, 2, 2, 3], [1, 5, 6, 3], [1, 3], [1]]

qiuyouzhi

沙雕写法

>>> lst = ''.join(map(str, l[1:])).split('1')
>>> res = [list(map(int, [*('1' + lst[i])])) for i in range(len(lst))]
>>> res
[[1, 2, 3], [1, 2, 2, 3], [1, 5, 6, 3], [1, 3], [1]]

qq1151985918

柿子饼同学 发表于 2021-4-11 20:00
没看懂

before 其实是我假设的 如果原始列表 lst 不是ls[0] == 1而做的
其他的就是加个标记 flg
首先数出 lst 有几个 1 就在 res 创建几个 空列表
然后每次  i == 1  就让 flg +1
就是这样

qq1151985918

柿子饼同学 发表于 2021-4-11 20:00
没看懂

我给你回复了，没显示出来，可能又在审核，你自己研究下吧。

before 的存在是我假设 如果 lst[0] 不等于1 而设置的

qq1151985918

qiuyouzhi 发表于 2021-4-11 20:15
沙雕写法

兄弟你这个不太靠谱吧？如果元素不是1位数而是2位数3位数呢？
l = [1, 22, 333, 1, 222, 2, 3, 1, 5, 6, 3, 1, 3, 1]
还行吗？
如果里面的元素包含1  比如101可还行？
l = [1, 22, 333, 101, 222, 2, 3, 1, 5, 6, 3, 1, 3, 1]

傻眼貓咪

arr = [1, 2, 3, 1, 2, 2, 3, 1, 5, 6, 3, 1, 3, 1]
res = []
if arr: temp = [arr[0]]

for n, i in enumerate(arr[1:], start = 1):
    if i == 1:
        res.append(temp)
        temp = [i]
    else:
        temp.append(i)
    if n == len(arr)-1:
        res.append(temp)

print(*res)

[1, 2, 3] [1, 2, 2, 3] [1, 5, 6, 3] [1, 3] [1]