从python列表中获取值序列

时间:2017-10-15 05:53:45

标签: python list itertools

我有一个像这样的数组,

a = [3,2,5,7,4,5,6,3,8,4,5,7,8,9,5,7,8,4,9,7,6]

我想制作小于7的值列表(如下所示)

b = [[3,2,5],[4,5,6,3],[4,5],[5],[4],[6]]

所以我使用了以下方法,

>>> from itertools import takewhile
>>> a = [3,2,5,7,4,5,6,3,8,4,5,7,8,9,5,7,8,4,9,7,6]
>>>list(takewhile(lambda x: x < 7 , a))
[3, 2, 5]

但我只得到第一个序列。任何人都可以帮我解决这个问题吗? 谢谢。

2 个答案:

答案 0 :(得分:6)

[[3, 2, 5], [4, 5, 6, 3], [4, 5], [5], [4], [6]]

输出:

const char*

答案 1 :(得分:3)

因为 takewhile 只要条件为True就返回元素,如果不是它就会中断,并保留该函数。

你需要这样的东西:

big_list = [].
small_list = []
for number in a:
    if number <7 :
        small_list.append(number)
    else:
        if small_list: # this is for not appending empty lists
            big_list.append(small_list)
        small_list = []