python - 打开元素包装并添加新元素

Question

我需要创建一个新列表来合并两个列表，其中一个是列表列表。这是我需要做的:

a = [[2, 1, 4, 5, 0], [3, 6, 5, 4, 8], [2, 1, 4, 7, 8], [3, 4, 9, 5, 6], [7, 5, 2, 1, 1]]
b = [2, 3, 5, 0, 8]
c = []

for indx, item in enumerate(a):
    c.append([item, b[indx]])

这生成 c 为:

[[[2, 1, 4, 5, 0], 2], [[3, 6, 5, 4, 8], 3], [[2, 1, 4, 7, 8], 5], [[3, 4, 9, 5, 6], 0], [[7, 5, 2, 1, 1], 8]]

但我需要它看起来像:

[[2, 1, 4, 5, 0, 2], [3, 6, 5, 4, 8, 3], [2, 1, 4, 7, 8, 5], [3, 4, 9, 5, 6, 0], [7, 5, 2, 1, 1], 8]

我试过在 item 前面添加一个 * 来解压元素，但这不起作用。

Answer 1

只需连接项目以从 item 和 b 中的元素创建一个新列表:

for indx, item in enumerate(a):
    c.append(item + [b[indx]])

您可以使用 zip() function 简化循环:

for a_item, b_item in zip(a, b):
    c.append(a_item + [b_item])

然后将 c 的整个定义移动到列表推导中:

c = [a_item + [b_item] for a_item, b_item in zip(a, b)]

演示:

>>> a = [[2, 1, 4, 5, 0], [3, 6, 5, 4, 8], [2, 1, 4, 7, 8], [3, 4, 9, 5, 6], [7, 5, 2, 1, 1]]
>>> b = [2, 3, 5, 0, 8]
>>> [a_item + [b_item] for a_item, b_item in zip(a, b)]
[[2, 1, 4, 5, 0, 2], [3, 6, 5, 4, 8, 3], [2, 1, 4, 7, 8, 5], [3, 4, 9, 5, 6, 0], [7, 5, 2, 1, 1, 8]]