有没有办法使用列表理解返回整数而不是列表？

如何解决有没有办法使用列表理解返回整数而不是列表？

我正在解决一个问题，我必须在列表中找到一个唯一的整数，这很容易使用 list.count(x) 解决。但是，我无法将代码压缩为一行。

def find_uniq(arr):
    return [x for x in arr if arr.count(x) == 1]

我的代码工作正常，但它返回例如：[2] 而不是 2，或 [0.55] 而不是 0.55。

有没有办法使用列表推导返回整数而不是包含整数的列表？

解决方法

不是生成 list 的 list comprehension，而是创建一个生成器，从中获取第一个元素：

return next(x for x in arr if arr.count(x) == 1)

这会引发 StopIteration 列表中没有元素满足条件；您可以改为返回一个默认值，例如 None，如下所示：

return next((x for x in arr if arr.count(x) == 1),None)

用count一次又一次地迭代数组是否明智也值得怀疑；根据其大小，这可能非常低效。先建立一个计数器可能更有效：

from collections import Counter

return next(v for v,c in Counter(arr).items() if c == 1)

,

对于列表理解，您已经有了答案——这是一种资源浪费。 但是，您的方法是有效的，因为它使用了日常/初学者结构。在同一个论点中，我想提出两点建议：

• 避免对 count() 的冗余调用；
• 避免创建列表（因为您只需要一次一个元素）。

假设我们有以下数组 arr：

> arr = [random.randint(0,9) for _ in range(10)]
> arr
[6,7,9,3,8,8]

对于第一点，您可以创建一个 set 来减少计数的数量：

> numbers_set = set(arr)
> numbers_set
{0,6,9}

然后你可以在我们朋友filter的帮助下拥有一个生成器：

> unique_numbers = filter(lambda x:arr.count(x)==1,numbers_set)
> print(next(unique_numbers))
0
> print(next(unique_numbers))
6
> print(next(unique_numbers))
7
> print(next(unique_numbers))
StopIteration:

,

如果您确定您只想从传递的列表中返回一个整数，您可以将返回更改为

def find_uniq(arr):
    return [x for x in arr if arr.count(x) == 1][0]

但它只会返回列表中唯一的第一个元素。如果你想返回更多的列表是更好的方法

,

使用索引返回。

def find_uniq(arr):
    return [x for x in arr if arr.count(x) == 1][0]