Python - 同时遍历两个列表而不在每次迭代中都移动到下一行

Question

我环顾四周并找到了 itertools 包，但据我所知，没有一个函数能完全满足我的需要。我已经开始自己编写一个函数，我可以做到，但我想知道是否有人知道可以完成全部或部分双列表迭代的内置函数？

我想做的是一次遍历两个列表，比较每个列表的值，根据比较的结果做一些事情，并可能在一个列表的同一行上移动到下一行第二个列表(基于比较的结果)。这是一个例子:

ListA= [[1, 7, 3],
        [1, 12, 4],
        [1, 9, 5]]

ListB= [[2, 2, 3],
        [2, 2, 3],
        [2, 5, 4]]

我想逐行浏览每个列表并比较最后位置的数字。如果它们相等，我想添加两个第二个数字。但只要 ListB 中的第三个数字等于 ListA 中的第三个数字，我就想继续添加它们。这意味着 ListA 可能会留在同一行，而 ListB 会向下移动几行(这就是 itertools 函数不起作用的原因，因为它们似乎都将每个列表的每一行串联起来)。所以我希望输出看起来像这样:

Iteration 1 ListOut= [[9, 3]]

Iteration 2 ListOut= [[11,3]]

Iteration 3 ListOut= [[11,3], [17, 4]]

Iteration 4 ListOut= [[11,3], [17, 4],  [9, 5]]

Answer 1

你想做这样的事情:

listb_iter = iter(ListB)
item_b = next(listb_iter) #we're assuming that ListB as at least one item.
for item_a in ListA:
    if ...: #condition to move on to the next item in ListB
        try:
            item_b = next(listb_iter)
        except StopIteration:
            #went through all of ListB

    #other logic