尝试 Python 使用两个不同范围的变量来理解列表

Question

我正在尝试使用大小为 n 和 n/2 的两个不同数组的内容快速生成一个列表。例如：

A = [70, 60, 50, 40, 30, 20, 10, 0]
B = [1, 2, 3, 4]

我希望生成类似

的东西

[(A[x], B[y]) for x in range(len(A)) for y in range(len(B))]

我了解第二个 for 语句是 "x" 语句之后的嵌套 for 循环。我试图让新数组的内容成为

A[0], B[0]
A[1], B[1]
A[2], B[2]
A[3], B[3]
A[4], B[0]
A[5], B[1]
A[6], B[2]
A[7], B[3]

谁能给我指出正确的方向？

Answer 1

不要使用嵌套循环；您正在配对 A 和 B，并根据需要重复 B。你需要的是zip() (to do the pairing), and itertools.cycle()（重复B）：

from itertools import cycle

zip(A, cycle(B))

如果 B 总是 A 的一半，你也可以将 B:

zip(A, B + B)

演示：

>>> from itertools import cycle
>>> A = [70, 60, 50, 40, 30, 20, 10, 0]
>>> B = [1, 2, 3, 4]
>>> zip(A, cycle(B))
[(70, 1), (60, 2), (50, 3), (40, 4), (30, 1), (20, 2), (10, 3), (0, 4)]
>>> zip(A, B + B)
[(70, 1), (60, 2), (50, 3), (40, 4), (30, 1), (20, 2), (10, 3), (0, 4)]

如果不知道哪一个是更长的列表，您可以使用 min() 和 max() 来选择循环哪一个：

zip(max((A, B), key=len), cycle(min((A, B), key=len))

或者对于要配对的任意数量的列表，将它们循环 all 但使用 itertools.islice() 将事物限制为最大长度：

inputs = (A, B)  # potentially more
max_length = max(len(elem) for elem in inputs)
zip(*(islice(cycle(elem), max_length) for elem in inputs))

演示：

>>> from itertools import islice
>>> inputs = (A, B)  # potentially more
>>> max_length = max(len(elem) for elem in inputs)
>>> zip(*(islice(cycle(elem), max_length) for elem in inputs))
[(70, 1), (60, 2), (50, 3), (40, 4), (30, 1), (20, 2), (10, 3), (0, 4)]

Answer 2

尝试只使用一个 for 循环而不是两个，并在超过其长度后让第二个回绕到 0。

[(A[x], B[x%len(B)]) for x in range(len(A))]

请注意，这仅在 A 是较长的列表时才有效。如果你知道 B 永远是 A 的一半，你也可以使用这个：

list(zip(A, B*2))

Answer 3

[(A[x % len(A)], B[x % len(B)]) for x in range(max(len(A), len(B)))]

无论 A 是否是更大的列表，这都有效。 :)

尝试 Python 使用两个不同范围的变量来理解列表

Attempting Python list comprehension with two variable of different ranges

python

list-comprehension

list