如何返回第一个项目，而不是整个列表，使用列表推导?

Question

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

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

我的代码工作得很好，但是它返回例如:[2]而不是2，或[0.55]而不是0.55。

是否有一种方法返回整数而不是包含整数的列表，使用列表推导式?

Answer 1

您已经有了列表理解的答案——这是对资源的浪费。不过，你的方法是有效的，因为它利用了日常/初学者的结构。在同样的论点中，我想提出两点建议:

• 避免冗余调用count();
• 避免创建列表(因为你每次只需要一个元素)。

假设我们有以下数组arr:

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

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

> numbers_set = set(arr)
> numbers_set
{0, 3, 6, 7, 8, 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:

Answer 2

不使用列表推导式生成列表，而是创建一个生成器，从中获取第一个元素:

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)

Answer 3

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

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

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

Answer 4

使用索引返回。

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