Question

我为这个令人困惑的头衔道歉。我想知道比较两个子列表列表的最佳方法是什么，如果子列表中的项目与另一个列表的子列表中的项目匹配，则前一个列表会被后者扩展＃＆## 39; s项目。我知道这听起来很混乱，所以这里有详细信息：

我有两个子列表列表：

listA = [['x', 'apple', 'orange'], ['y', 'cat', 'dog'], ['z', 'house', 'home']]
listB = [['z', 7, 8, 9], ['x', 1, 2, 3], ['y', 4, 5, 6]]

现在，如果listA的子列表中的第一项与listB的项匹配，我希望扩展listA，使其包含listB中的值的子列表。基本上，最终结果应该如下：

listA = [['x', 'apple', 'orange', 1, 2, 3], ['y', 'cat', 'dog', 4, 5, 6], ['z', 'house', 'home', 7, 8, 9]]

以下是我尝试的内容：

for (sublistA, sublistB) in zip(listA, listB):
    if sublistA[0] == sublistB[0]:
        sublistA.extend(sublistB[1], sublistB[2], sublistB[3])

然而，似乎代码在if语句中失败了。当我打印listA时，我得到的只是它的原始项目：

>>> print(listA)
[['x', 'apple', 'orange'], ['y', 'cat', 'dog'], ['z', 'house', 'home']]

为什么if语句不起作用？有什么方法可以进行匹配，然后提取项目？

编辑：根据idjaw的建议，我创建了第三个列表并尝试再次执行上述操作。但是，我似乎回到了一个空列表，因为if语句似乎不再起作用。这是代码：

listC = []
for (sublistA, sublistB) in zip(listA, listB):
    if sublistA[0] == sublistB[0]:
        listC.append(sublistA[0], sublistA[1], sublistA[2], 
                     sublistB[1], sublistB[2], sublistB[3])
print(listC)

输出：[]

Answer 1

这是通过构建一个dict来实现这一目的，以便更容易地查找要添加到的列表：

代码：

lookup = {x[0]: x for x in listA}
for sublist in listB:
    lookup.get(sublist[0], []).extend(sublist[1:])

测试代码：

listA = [['x', 'apple', 'orange'], ['y', 'cat', 'dog'], ['z', 'house', 'home']]
listB = [['z', 7, 8, 9], ['x', 1, 2, 3], ['y', 4, 5, 6]]

lookup = {x[0]: x for x in listA}
for sublist in listB:
    lookup.get(sublist[0], []).extend(sublist[1:])

print(listA)

结果：

[
    ['x', 'apple', 'orange', 1, 2, 3], 
    ['y', 'cat', 'dog', 4, 5, 6], 
    ['z', 'house', 'home', 7, 8, 9]
]

Answer 2

也许你的代码可能就是这个

listA = [['x', 'apple', 'orange'], ['y', 'cat', 'dog'], ['z', 'house', 'home']]
listB = [['z', 7, 8, 9], ['x', 1, 2, 3], ['y', 4, 5, 6]]



for la in listA:
    for lb in listB:
        if la[0] == lb[0]:
            for i in lb[1:]:
                la.append(i)

print(listA)

Python从子列表中提取项目，如果它们与另一个列表的子列表中的项目匹配

2 个答案:

代码：

测试代码：

结果：