根据条件删除python集中的条目

Question

我使用了 scipy.spatial.KDTree.query_pairs() ，它返回了一组 python 元组。 假设这是输出：

set1 = {(2, 3), (4, 5), (1, 6), (6, 7), (3, 8), (6, 8)}

接下来，我要删除集合中所有不满足条件的元组

arr = [6, 7]
tuple[0] in arr or tuple[1] in arr

最优雅/快速/pythonic 的方式是什么？我应该转换为 numpy 数组吗？

Answer 1

您需要遍历并检查set1 中的每个元组，您可以使用set comprehension 和any() 来做到这一点：

>>> set1 = {(2, 3), (4, 5), (1, 6), (6, 7), (3, 8), (6, 8)}
>>> arr = [6, 7]
>>> set2 = set(arr)  # convert to set for O(1) lookup time
>>> set3 = {t for t in set1 if any(x in set2 for x in t)}
>>> set3
{(6, 7), (6, 8), (1, 6)}

Answer 2

解决此问题的一种方法是遍历集合并为满足条件的元素创建一个数组，然后将该数组转换为一个集合。

这可以通过单行解决方案来完成，例如：

set2 = set([a for a in set1 if a[0] not in key and a[1] not in key])

Answer 3

希望这就是你要找的东西

filtered = [x,y for x,y in set1 if (x,y) == (6,7)]

或尝试将 lambda 用于复杂的过滤器。

Answer 4

如果不将您的数据转换为 numpy 数组，这就是我解决此问题的方法：

set1 = {(2, 3), (4, 5), (1, 6), (6, 7), (3, 8), (6, 8)}
arr = [6, 7]

filtered = list(filter(lambda xy: bool(set(xy) & set(arr)), set1))

如果 set1 中元素的长度显着增长并且您决定使用 numpy，那么我只需更改 numpy 的集合交集>的intersect1d

如果 set1 的长度反而增加了，那么我建议你看看另一个 StackOverflow post 看看如何过滤一个 numpy 数组。