2013-05-03 33 views
0

我有两个列表,它们基本上需要根据它们的匹配项目(列表)相互映射。输出是已映射的对的列表。当要映射的列表长度为一时,我们可以在另一个列表中查找直接匹配。问题出现,当被映射列表的长度为> 1,我需要找到,如果在一个列表是B.在Python中将两个列表根据其项目映射到列表对中

的一个子集

输入:

A = [['point'], ['point', 'floating']] 
B = [['floating', 'undefined', 'point'], ['point']] 

我的失败代码:

C = [] 
for a in A: 
    for b in B: 
     if a == b: 
      C.append([a, b]) 
     else: 
      if set(a).intersection(b): 
       C.append([a, b]) 

print C 

预期输出:

C = [ 
    [['point'], ['point']], 
    [['point', 'floating'], ['floating', 'undefined', 'point']] 
    ] 
+0

为什么'[['point','floating'],['point']]'不在预期的输出中? – 2013-05-03 19:52:08

+0

@ashwini请参阅上面的说明。我已经解释了Y. – Shankar 2013-05-03 19:56:35

+0

“长度> 2”(您现在更新了)正在窃听我,请参阅下面的解决方案。 – 2013-05-03 20:09:55

回答

1

只需添加一个长度条件为elif声明:

import pprint 
A = [['point'], ['point', 'floating']] 
B = [['floating', 'undefined', 'point'], ['point']] 
C = [] 

for a in A: 
    for b in B: 
     if a==b: 
      C.append([a,b]) 
     elif all (len(x)>=2 for x in [a,b]) and not set(a).isdisjoint(b): 
      C.append([a,b]) 

pprint.pprint(C) 

输出:

[[['point'], ['point']], 
[['point', 'floating'], ['floating', 'undefined', 'point']]] 
1

只是为了利益的缘故,这里的 “一条线” 的实施使用itertools.ifilter

from itertools import ifilter 

C = list(ifilter(
    lambda x: x[0] == x[1] if len(x[0]) == 1 else set(x[0]).issubset(x[1]), 
    ([a,b] for a in A for b in B) 
)) 

编辑:

有阅读问题的最新评论,我想我可能误解究竟被认为是匹配的。在这种情况下,这样的事情可能会更合适。

C = list(ifilter(
    lambda x: x[0] == x[1] if len(x[0])<2 or len(x[1])<2 else set(x[0]).intersection(x[1]), 
    ([a,b] for a in A for b in B) 
)) 

无论哪种方式,基本概念是相同的。只需改变lamba中的条件,就可以完全匹配你想要匹配的东西。

相关问题