将一维数组的元素插入到二维数组中的特定位置

Question

尝试使用insert() 方法组合包含字符串的一维和二维列表/数组。

但是，从 1D 列表中获取特定元素并将其放置到 2D 列表中的特定位置是我卡住的地方。

这是目标的简化版本；

#2D list/array
list1= [['a1','b1'], ['a2','b2'] , ['a3','b3']]

#1D list/array
list2= ['c3','c2','c1']

#desired output
list1= [['a1','b1','c1'], ['a2','b2','c2'] , ['a3','b3','c3']]

这是我尝试使用的脚本中的隔离代码块；

#loop through 1D list with a nested for-loop for 2D list and use insert() method.
#using reversed() method on list2 as this 1D array is in reverse order starting from "c3 -> c1"
#insert(2,c) is specifying insert "c" at index[2] location of inner array of List1

for c in reversed(list2):
    for letters in list1:
        letters.insert(2,c)

print(list1)

以上代码的输出；

[['a1', 'b1', 'c3', 'c2', 'c1'], ['a2', 'b2', 'c3', 'c2', 'c1'], ['a3', 'b3', 'c3', 'c2', 'c1']] 

返回所需输出的最佳和最有效的方法是什么？我应该使用append() 方法而不是insert() 还是应该在使用任何方法之前引入列表连接？

任何见解将不胜感激！

Answer 1

正如 cmets 中所讨论的，您可以通过使用 enumerate 或 zip 的列表推导来实现此目的。您可以使用enumerate 从list1 获取索引和子列表，使用index 从list2 中选择适当的值附加到每个子列表：

list1 = [l1 + [list2[-i-1]] for i, l1 in enumerate(list1)]

或者你可以zip一起list1和相反的list2：

list1 = [l1 + [l2] for l1, l2 in zip(list1, list2[::-1])]

或者你可以使用一个简单的for循环来修改list1：

for i in range(len(list1)):
    list1[i].append(list2[-i-1])

对于所有这些，输出是：

[['a1', 'b1', 'c1'], ['a2', 'b2', 'c2'], ['a3', 'b3', 'c3']]