0

我有两个列表列表，基本上需要根据它们的匹配项（列表）相互映射。输出是映射对的列表。当要映射的列表长度为 1 时，我们可以在另一个列表中查找直接匹配项。问题出现了，当要映射的列表长度 > 1 时，我需要在其中找到，如果 A 中的列表是 B 的子集。

输入： 

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

2 回答 2

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 中的条件以完全匹配您想要匹配的内容。

于 2013-05-03T20:50:12.277 回答
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']]]
于 2013-05-03T19:44:41.553 回答