如果列表中的语句与lambda一起理解 [英] If statement in list comprehension with lambda

问题描述

我有

listName = [[0,1,2,15,16,17,2,3,4,6,8,9]]

我的代码行

[list(g) for k, g in groupby(listName, key=lambda i,j=count(): i-next(j))]

将 listName 拆分为 [[0,1,2]，[15,16,17]，[2,3,4]，[6,8,9]]
我希望分裂发生仅当下一个数字小于前一个数字时。 ie
我希望我的 listName 分成

[[0,1,2,15,16,17],[2,3,4,6,8,9]]

谢谢！ ：）

推荐答案

使用 itertools.chain 创建一个迭代器并展平你的列表：

It is much easier to use a generator function, using itertools.chain to create an iterator and flatten your list:

listName = [[0, 1, 2, 15, 16, 17, 2, 3, 4, 6, 8, 9]]

from itertools import chain
def split(l):
    it = chain(*l)
    prev = next(it)
    tmp = [prev]
    for ele in it:
        if ele < prev:
            yield tmp
            tmp = [ele]
        else:
            tmp.append(ele)
        prev = ele
    yield tmp


print(list(split(listName)))

这篇关于如果列表中的语句与lambda一起理解的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！