循环遍历列表直到找到值然后中断？

Question

我有一个列表：[1,2,3,4,5,6,0,9,2]，我想循环访问它。

我想为每 3 个元素创建另一个字典列表，直到达到 0 元素（即第一个字典将是 {'gold':1,'silver':2,'bronze':3} 和 4,5,6 类似）。循环将在到达0 后终止。

所以结果列表将是：[{'gold':1,'silver':2,'bronze':3},{'gold':4,'silver':5,'bronze':6}]

作为 Python 新手并具有 PHP 背景，我将如何利用 zip 或 map 等函数解决此问题，或者我是否必须求助于普通的 while 循环？

Answer 1

from itertools import cycle

placings = cycle(['gold', 'silver', 'bronze'])
numbers = [1,2,3,4,5,6,0,9,2]

def zipper(numbers, placings):

    d = {}

    for n, c in zip(numbers, placings):

        if n == 0:
            break

        d[n] = c

        if n % 3 == 0:
            yield d
            d = {}

print list(zipper(numbers, placings))

Answer 2

from itertools import takewhile, islice


def groups(l, keys, i, n):
    # lambda x: x != 0 -> take elements until we encounter any 0
    tk = takewhile(lambda x: x != i, l)
    while True:
        d = dict(zip(keys, islice(tk, n)))
        if len(d) == n: # make sure we have n keys in the dict
            yield d
        else: 
            break

keys = ['gold', 'silver', 'bronze']
print(list(groups(l,keys,0,3)))
[{'bronze': 3, 'gold': 1, 'silver': 2}, {'bronze': 6, 'gold': 4, 'silver': 5}]

takewhile

创建一个迭代器，只要谓词为真，它就会从可迭代对象中返回元素。

所以我们只取元素直到我们遇到的第一个 i，d = islice(tk,n) 使用 islice 从我们的 takewhile 对象中取一个 n 长度切片，然后我们从传入的键创建一个字典，创建dict 一次消耗n 元素，因此每次调用它时我们都会移动到接下来的三个元素。如果 takewhile 对象的长度不是n 的倍数，那么我们最后会有一个小于n 大小的dict，所以if len(d) == n: 会捕捉到它并打破循环，所以奇怪的是长度字典或空字典将终止while。

如果您允许使用少于三个键的字典：

from itertools import takewhile, islice,chain

def groups(l, keys, i, n):
    tk = takewhile(lambda x: x != i, l)
    return (dict(zip(keys, chain(islice(tk, n - 1),(i,)))) for i in tk)

输出：

In [5]: l = [1, 2, 3, 4, 5, 6, 7, 11, 6, 9, 0, 2]

In [6]: list(groups(l,['gold', 'silver', 'bronze'],0,3))
Out[6]: 
[{'bronze': 1, 'gold': 2, 'silver': 3},
 {'bronze': 4, 'gold': 5, 'silver': 6},
 {'bronze': 7, 'gold': 11, 'silver': 6},
 {'gold': 9}]

islice(tk, n - 1) 从我们的 takewhile 对象 len n - 1 中获取一个切片，我们将它和i 链接起来，所以我们每次都得到n 元素。

或者使用filter过滤掉少于3个key的dict：

def groups(l, keys, i, n):
    tk = takewhile(lambda x: x != i, l)
    return filter(lambda x: len(x) == n, (dict(zip(keys, chain(islice(tk, n - 1),(i,)))) 
                                          for i in tk))

删除较短的字典：

In [8]: list(groups(l,['gold', 'silver', 'bronze'],0,3))
Out[8]: 
[{'bronze': 1, 'gold': 2, 'silver': 3},
 {'bronze': 4, 'gold': 5, 'silver': 6},
 {'bronze': 7, 'gold': 11, 'silver': 6}]

这与之前的相同，我们只是从末尾过滤任何可能的奇数长度字典。对于 python2 使用 itertools.ifilter 而不是过滤器。

Answer 3

简单！用一个简单的函数，一个数据结构和简单的迭代：

示例：

from collections import defaultdict


def f(xs, labels=("gold", "silver", "bronze")):
    d = defaultdict(int)
    n = 0
    for i, x in enumerate(xs):
        if x == 0:  # break if we see a 0
            return d

        d[labels[n]] += i  # update our counts

        # cycle through gold/silver/bronze (labels)
        n += 1
        if n % 3 == 0:
            n = 0

    return d


xs = [1, 2, 3, 4, 5, 6, 0, 9, 2]
d = f(xs)
print d

输出：

$ python foo.py 
defaultdict(<type 'int'>, {'bronze': 7, 'silver': 5, 'gold': 3})

Answer 4

首先，您需要将较大的列表分成与列表medals 相同大小的较小列表，以使zip() 函数正常工作，并且0 在列表中的位置应位于位置可以被medals 列表的长度整除，否则较小列表中的最后一个列表将具有较少数量的元素。

listOfDictionary = []
lst = [1,2,3,4,5,6,0,9,2]

ind = lst.index(0)   #Tells where to break the iteration loop.

medals = ["gold", "silver", "bronze"]

if ind%3==0:

    newlst = [lst[i: i+3]for i in xrange(0,ind,3)]
    #newlst = [[1, 2, 3], [4, 5, 6]]

    for lsst in newlst:
        listOfDictionary.append(dict(zip(medals, lsst)))


print listOfDictionary
>>> [{'bronze': 3, 'silver': 2, 'gold': 1}, {'bronze': 6, 'silver': 5, 'gold': 4}]

Answer 5

在列表的情况下使用列表推导然后使用循环

In [61]: mykeys
Out[61]: [1, 2, 3, 4, 5, 6, 0, 9, 2]

In [63]: mykeys1=[i for i in mykeys[:-1 if 0 not in mykeys else mykeys.index(0)]]

In [64]: myvalues
Out[64]: ['gold', 'silver', 'bronze']

In [66]: myvalues1=cycle(myvalues)

In [67]: mydict=dict(zip(mykeys1,myvalues1))

In [68]: mydict
Out[68]: {1: 'gold', 2: 'silver', 3: 'bronze', 4: 'gold', 5: 'silver', 6: 'bronze'}
In [69]: [dict(mydict.items()[i:i+3]) for i in range(0,len(mykeys1),3)]
Out[69]: [{1: 'gold', 2: 'silver', 3: 'bronze'}, {4: 'gold', 5: 'silver', 6: 'bronze'}]