基于迭代的内部列表值合并两个外部列表

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']]

---

在 cmets 中，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 更方便的少数几次之一。

Answer 2

如果你想将[[1, 'a'], [2, 'b']] 和[[1, 'c'], [3, 'd']] 合并到[[1, 'a', 'c'], [2, 'b'], [3, 'd']]：

from collections import defaultdict
dict1_2 = defaultdict(list)
dict1_2.update((item[0], item[1:]) for item in list1)
for item in list2:
    dict1_2[item[0]].append(item[1:])

如果你想让它们合并到[[1, 'a', 'c']]:

dict1 = dict((item[0], item[1:]) for item in list1)
dict1_2 = {}
for item in list2:
    key = item[0]
    if key in dict1:
        dict1_2[key] = dict1[key] + item[1:]

您使用item[0] 作为键，因此您应该使用适合它的数据类型。在这种情况下，这是一个字典/映射。

这（平均而言）在线性时间内起作用，O(m+n)（其中 m 和 n 是列表的长度）。任何使用嵌套循环或类似的解决方案都将是 O(m*n)

如果您确实需要将数据作为列表返回，您可以这样做

list1_2 = [[key] + value for key, value in dict1_2.iteritems()]

Answer 3

resultlist = []
for x in list1:
    for y in list2:
        if x[0] == y[0]:
            resultlist.append(x+y)

Answer 4

不是最好的方法，但绝对简洁且难以阅读，如果你想要的话：

>>> l1 = [['1', 'expired', 'test', '0'], ['31', 'active', 'test', '1']]
>>> l2 = [['1', 'Andrew', 'Alexander'], ['31', 'John', 'Smith']]

>>> [sl1 + list(*[sl2[1:] for sl2 in l2 if sl2[0]==sl1[0]]) for sl1 in l1]

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

请不要在任何实际代码中实际使用它。

Answer 5

l1 = [['1', 'expired', 'test', '0'], ['31', 'active', 'test', '1']]
l2 = [['1', 'Andrew', 'Alexander'], ['31', 'John', 'Smith'], ['51', 'Johnny', 'Nomatch']]

from itertools import groupby, chain
from operator import itemgetter

all = sorted(chain(l1,l2), key=itemgetter(0)) # puts the related lists together
groups = groupby(all, itemgetter(0)) # groups them by first element
chains = (group for key, group in groups) # get each group
print [list(chain.from_iterable(g)) for g in chains] # merge them

这是一个单线；-)

包含不匹配的项目。你可以通过简单地检查len(group) > 4来过滤掉它们。