python - 遍历两个不同长度的列表

Question

我有 2 个长度不同的数字列表，例如:

list1 = [1, 2, -3, 4, 7]
list2 = [4, -6, 3, -1]

我需要用函数迭代这些:

final_list = []
for index in range(???):
    if list1[index] < 0:
        final_list.insert(0, list1[index])
    elif list1[index] > 0:
        final_list.insert(len(final_list), list1[index])
    if list2[index] < 0:
        final_list.insert(0, list2[index])
    elif list2[index] > 0:
        final_list.insert(len(final_list), list2[index])
return final_list

但无法弄清楚如何处理范围，因为如果我使用 max 长度，较短的列表将变得“超出范围”。关于如何克服这个问题或如何改变我的功能有什么想法吗？

Answer 1

itertools.zip_longest(*iterables, fillvalue=None)将为您完成这项工作:

If the iterables are of uneven length, missing values are filled-in with fillvalue.

对于您的示例列表，这将产生:

>>> import itertools
>>> list1 = [1, 2, -3, 4, 7]
>>> list2 = [4, -6, 3, -1]

>>> for combination in itertools.zip_longest(list1, list2):
    print(combination)

(1, 4)
(2, -6)
(-3, 3)
(4, -1)
(7, None)

---

如果您只想使用两个 列表中存在的值，请使用内置的 zip() :

The iterator stops when the shortest input iterable is exhausted.

>>> for combination in zip(list1, list2):
    print(combination)

(1, 4)
(2, -6)
(-3, 3)
(4, -1)