Question

我有一个包含字符串和另一个列表的元组列表。它如下：

board = [('10S', []), ('3H', []), ('6C', ['KS', '9C']), ('6H', []), ('7D', []), ('9S', ['AS', 'KS'])]

我想遍历这样的比较，如下所示：

10S 3H
3H 6C
6C KS
6C 9C
6C 6H
6H 7D

依旧......

下面写的get_mapping()会调用prev2和prev1的值。

我已经实现了以下代码：

for val in board:
    print "Val is: " + str(val)
    if cnt == 0:
        prev2 = val[0][0]
        prev1 = val[1][0]
        cnt += 1
        get_mapping(prev2,prev1,True)
    else:
        prev2 = prev1
        if len(val[1]) > 0:
            for v in val[1]:
                prev1 = v
                cnt += 1
                get_mapping(prev2,prev1,False)
        else:
            prev1 = val[0]
            cnt += 1
            get_mapping(prev2,prev1,True)

我知道第一个条件是错误的。关于如何遍历以进行比较，我有点困惑。任何帮助将不胜感激。

Answer 1

你可以编写一个产生对的生成器：

from itertools import zip_longest

master_board_state = [('10S', []), ('3H', []), ('6C', ['KS', '9C']),
                      ('6H', []), ('7D', []), ('9S', ['AS', 'KS'])]

def pairs(state):
    for (key, lst), nxt in zip_longest(state, state[1:]):
        yield from ((key, x) for x in lst)
        if nxt is not None:
            yield key, nxt[0]

print('\n'.join(str(p) for p in pairs(master_board_state)))

输出：

('10S', '3H')
('3H', '6C')
('6C', 'KS')
('6C', '9C')
('6C', '6H')
('6H', '7D')
('7D', '9S')
('9S', 'AS')
('9S', 'KS')

在上面zip_longest中，state会产生成对的项目。由于第一个参数比第二个参数长，因此最后一次迭代None用作fillvalue。然后循环将首先返回yield from元组的(key, list item)生成器，最后生成当前和下一个键。

更新对于Python 2，需要进行少量修改，因为它没有yield from而zip_longest被称为izip_longest：

from itertools import izip_longest

master_board_state = [('10S', []), ('3H', []), ('6C', ['KS', '9C']),
                      ('6H', []), ('7D', []), ('9S', ['AS', 'KS'])]

def pairs(state):
    for (key, lst), nxt in izip_longest(state, state[1:]):
        for x in lst:
            yield key, x
        if nxt is not None:
            yield key, nxt[0]

print '\n'.join(str(p) for p in pairs(master_board_state))

Answer 2

所以这是一种非常特殊的遍历，可以通过生成器完成。

更新：最初尝试使用复杂的生成器表达式但是错过了尾部，所以不得不最终得到类似于@niemmi的生成器函数，但方法略有不同。

from itertools import tee

# from itertools recipes
def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return zip(a, b)

def pairs(d):
    for (a, x), (b, y) in pairwise(d):
        yield from ((a, n) for n in x)
        yield a, b
    yield from ((b, n) for n in y)

>>> list(pairs(board))
[('10S', '3H'),
 ('3H', '6C'),
 ('6C', 'KS'),
 ('6C', '9C'),
 ('6C', '6H'),
 ('6H', '7D'),
 ('7D', '9S'),
 ('9S', 'AS'),
 ('9S', 'KS')]

遍历包含python中的字符串和列表的元组列表

2 个答案: