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

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

（这比张贴在原来的问题所需的输出不同。）

Then：

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 

这是一个oneliner ;-)

项不匹配的也包括在内。您可以通过简单地检查len(group) > 4来过滤它们。