如何从列表中返回重复序列

Question

我正在尝试从列表中返回一组重复出现的数字。我想我错过了什么......

list = range(0,10)
K = (1,4)

def f(x):
    for k in K:
        yield [i for i in x if i <= 1+k or i >= 4+k]       

print filter(f, list)

我希望可以为定义设置一个循环。但是输出是[0, 1, 2, 5, 6, 7, 8, 9]，显然不是预期的[0,1,2,5,8,9]。

那么如何分离值呢？

Answer 1

首先，您的代码需要更正如下：

def f(x):
    for k in K:
        yield [i for i in x if i <= 1 + k or i >= 4 + k]

通过此更正，list(f(range(0, 10))) 生成 [[0, 1, 2, 5, 6, 7, 8, 9], [0, 1, 2, 3, 4, 5, 8, 9]]。

下一步是过滤两个集合的交集元素。这可以通过多种方式完成，如问题Find intersection of two lists? 所示。

这是我的首选方法，基于this answer：

def intersect(lists):
    count = len(lists)

    if count == 0:
        return []
    elif count == 1:
        return lists
    elif count == 2:
        es1 = set(lists[1])
        return [x for x in lists[0] if x in es1]
    else:
        return intersect((lists[0], intersect(lists[1:])))

现在，使用 intersect 函数在问题中编写的函数的更正版本，您将获得预期的结果，如下所示：

result = list(f(range(0, 10)))
result
# [[0, 1, 2, 5, 6, 7, 8, 9], [0, 1, 2, 3, 4, 5, 8, 9]]
intersect(result)
# [0, 1, 2, 5, 8, 9]

---

警告！！

潜在陷阱：
Python 将允许您为变量指定标准函数的名称。但是，这样做会隐藏您使用的名称的函数，使函数隐藏并可能导致意外行为。因此，建议您将 list 变量的名称更改为其他名称。例如，您可以尝试l1 或list1。