python - 根据迭代的内部列表值合并两个外部列表

Question

我有两个列表列表 - 即

[['1', 'expired', 'test', '0'], ['31', 'active', 'test', '1']]

以及

[['1', 'Andrew', 'Alexander'], ['31', 'John', 'Smith']]

我们称它们为list1和list2

我想合并 list1 和 list2，但仅当(注意，这是伪代码，试图弄清楚如何在 Python 中对其进行编程)

x[0] in list1 == x[0] in list2

我不知道如何写出来。

我所说的合并是指(伪代码)

list[x] = list1[x] + list2[x] while x[0] in list1 == x[0] in list2

所需输出:

[['1', 'expired', 'test', '0', '1', 'Andrew', 'Alexander'], ['31', 'active', 'test', '1', '31', 'John', 'Smith']]

唯一的关键点是并非所有 x[0] 都会完美匹配。

Answer 1

利用agf的想法，采用collections.defaultdict ，这在 O(m+n) 中，其中 m 和 n 是列表的长度。

import collections
import itertools

x=[['1', 'expired', 'test', '0'], ['31', 'active', 'test', '1']]
y=[['1', 'Andrew', 'Alexander'], ['31', 'John', 'Smith']]

result=collections.defaultdict(list)
for item in itertools.chain(x,y):
    result[item[0]].append(item)
result=[list(itertools.chain.from_iterable(value)) for value in result.values()]
print(result)

产量

[['1', 'expired', 'test', '0', '1', 'Andrew', 'Alexander'], ['31', 'active', 'test', '1', '31', 'John', 'Smith']]

<小时/>

在评论中OP说期望的输出是

[['1', 'expired', 'test', '0', 'Andrew', 'Alexander'], ['31', 'active', 'test', '1', 'John', 'Smith']]

(这与原始问题中发布的所需输出不同。)

然后:

import collections
import itertools

x=[['1', 'expired', 'test', '0'], ['31', 'active', 'test', '1']]
y=[['1', 'Andrew', 'Alexander'], ['31', 'John', 'Smith']]

result={}
for item in itertools.chain(x,y):
    result.setdefault(item[0],item[:1]).extend(item[1:])
result=result.values()
print(result)

这是我发现使用 setdefault 的少数几次之一比collections.defaultdict更方便。