Question

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

然而，Im坚持要从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

如评论中所述，您可以使用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']]

对于所有这些，输出为：

{{1}}

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

1 个答案: