如何使用 yield 将两个列表中的数字按索引连接到一个元组中？（Python）

Question

我需要制作一个“生成器函数”，它将获取 2 个列表并将两个列表中的数字按索引连接到一个元组中。 例如：

l1 = [3, 2, 1]
l2 = [4, 3, 2]

第一次迭代的结果将是(3, 4)第二次迭代的结果将是(2, 3)而第三次迭代(1, 2)而且，一个列表可能比另一个列表有更多的数字。 在那种情况下，我需要编写条件“如果其中一个列表在迭代时结束，则不再执行进一步的迭代。” （使用try, except ）

我知道，生成器函数使用yield而不是return ，但我不知道如何编写这个函数......我这样做了

def generate_something(l1, l2):
    l3 = tuple(tuple(x) for x in zip(l1, l2))
    return l3

output 是

((3, 4), (2, 3), (1, 2))

它有效，但这不是 geterator function，没有yield ，没有第一、第二等迭代。 我希望你能帮帮我...

Answer 1

可能对您有帮助：

def generate_something_2(l1, l2):
    # if one of the lists ended while iterating, then no further iterations are performed.
    # For this get minimal len of list
    min_i = len(l1) if len(l1) < len(l2) else len(l2)
    for i in range(0,min_i):
        yield (l1[i], l2[i])

l1 = [3, 2, 1, 1]
l2 = [4, 3, 2]

l3 = tuple(generate_something_2(l1, l2))
# Result: ((3, 4), (2, 3), (1, 2))
print(l3)

Answer 2

您可以创建自己的生成器：

def iterate_lists(l1,l2):
   for i,item in enumerate(l1):
       yield l1[i],l2[i]

或者作为生成器理解的变量：

iterate_lists = (li[i],l2[i] for i,item in enumerate(l1))

关于如果长度不相等，我不清楚你到底想做什么，如果没有消息，你只能在较小的列表中使用 go...只需将for i,item in enumerate(l1)更改为for i in range(min((len(l1),len(l2)))

Answer 3

使用 while 循环并跟踪索引 position，当出现IndexError时，您已经到达最短列表的末尾并停止：

def generate_something(l1, l2):
    idx = 0
    while True:
        try:
            yield l1[idx], l2[idx]
            idx += 1
        except IndexError:
            break


l1 = [3, 2, 1]
l2 = [4, 3, 2]

g = generate_something(l1, l2)
print(list(g))

Output：

[(3, 4), (2, 3), (1, 2)]