Question

在python中，我有两个列表A和B。两个列表都包含元组(x,y)。例如：

A = [('x1','y1'), ('x2','y2'), ('x3','y3')]
B = [('x1','y1'), ('x2','y5'), ('x4','y4')]

现在，我想要三个结果。如果只涉及没有元组，所有这些都可以用集合论容易地解决。

结果1：两个列表的交叉点：set(A) & set(B)）。因此结果应该是比较两个列表的元组的两个值。结果应为：C = [('x1','y1')]

结果2：只有(x,y)[0]匹配的两个列表的交集。结果应为：D = (('x1','y1'), ('x2', ('y2', 'y5'))]。理想情况下，解决方案是D - C -> E = [('x2', ('y2', 'y5'))]，但我可以忍受D本身。

结果3：列表B的唯一身份与A：set(B)-(set(A) & set(B))相比较。仅在(x,y)[0]进行比较。结果应为：[('x4', 'y4')]。

我找不到任何关于这些问题的内容，也无法自己构建解决方案。有人可以帮忙吗？

Answer 1

为什么不使用python＆＃39; s set()？ 1是非常直接的，2是需要更多的工作：

A = [('x1','y1'), ('x2','y2'), ('x3','y3')]
B = [('x1','y1'), ('x2','y5'), ('x4','y4')]

a,b = set(A),set(B)
print '1:',a&b

axs = set(map(itemgetter(0),A))
bxs = set(map(itemgetter(0),B))

result2 = []
for c in axs&bxs:
    result2.append((c,set([y for x,y in A+B if x==c]))
print '2:',result2

输出：

1: set([('x1', 'y1')])
2: [('x2', set(['y2', 'y5'])), ('x1', set(['y1']))]

你可以使用非常类似的方法来实现3

Answer 2

以下是使用dicts执行所需操作的一些方法。这是Python 2代码;它需要对Python 3进行一些小修改.IIRC，Python 3没有dict.iteritems()，因为它的dict.items()返回迭代器而不是列表。

A = [('x1','y1'), ('x2','y2'), ('x3','y3')]
B = [('x1','y1'), ('x2','y5'), ('x4','y4')]

dA = dict(A)
dB = dict(B)

#Intersection, the simple way
print 'Result 1a:', list(set(A) & set(B))

#Intersection using dicts instead of sets
result = [(k, vA) for k, vA in dA.iteritems() if dB.get(k) == vA]
print 'Result 1b:', result

#match on 1st tuple element, ignoring 2nd element
result = {}
for k, vA in dA.iteritems():
    vB = dB.get(k)
    if vB:
        result[k] = (vA, vB) if vB != vA else vA
print 'Result 2a:', result.items()

#match on 1st tuple element only if 2nd elements don't match
result = {}
for k, vA in dA.iteritems():
    vB = dB.get(k)
    if vB and vB != vA:
        result[k] = (vA, vB)
print 'Result 2b:', result.items()

#unique elements of B, ignoring 2nd element
result = [(k, vB) for k, vB in dB.iteritems() if k not in dA]
print 'Result  3:', result

<强>输出

Result 1a: [('x1', 'y1')]
Result 1b: [('x1', 'y1')]
Result 2a: [('x2', ('y2', 'y5')), ('x1', 'y1')]
Result 2b: [('x2', ('y2', 'y5'))]
Result  3: [('x4', 'y4')]

Answer 3

两个列表的交集：

您已经知道解决方案：set(A) & set(B)。或者，相当于set(A).intersection(B)。

>>> A = [('x1', 'y1'), ('x2', 'y2'), ('x3', 'y3')]
>>> B = [('x1', 'y1'), ('x2', 'y5'), ('x4', 'y4')]
>>> set(A).intersection(B)
{('x1', 'y1')}

两个列表的交点，其中只有（x，y）[0]匹配：

首先，确保A和B都按x坐标排序。

然后使用itertools.groupby()和字典：

>>> a_grouped = {x: list(points) for x, points in
...              itertools.groupby(A, lambda point: point[0])}
>>> b_grouped = {x: list(points) for x, points in
...              itertools.groupby(B, lambda point: point[0])}
>>> [(x, {point[1] for point in a_grouped[x] + b_grouped[x]})
...  for x in a_grouped if x in b_grouped]
[('x2', {'y5', 'y2'}), ('x1', {'y1'})]

（这与你提出的不完全相同，因为正如你所看到的，我们有('x1', {'y1'})而不是('x1', 'y1')。此外，我们设置了而不是列表，但这些都是一些微不足道的事情。）

如果您要排除公共点：请在调用groupby（）之前将其从A和B中删除：

>>> A = set(A)
>>> B = set(B)
>>> common_points = A & B
>>> A = [point for point in A if point not in common_points]
>>> B = [point for point in B if point not in common_points]

列表B的唯一标记与A进行比较，仅在(x,y)[0]进行比较：

构造A中的点的所有x坐标的集合：
```
>>> exclude = {point[0] for point in A}
>>> [point for point in B if point[0] not in exclude]
[('x4', 'y4')]
```
请注意exclude的元素是a_grouped的键 - 这意味着您可以重用上一个问题中的部分代码并写入：
```
>>> [point for point in B if point[0] not in a_grouped]
[('x4', 'y4')]
```

对于所有这些解决方案，可以提高性能和可读性，如果您要使用我的代码，请考虑这一点。

在元组列表上设置理论魔力

3 个答案: