如何检查列表是否只包含特定项

Question

我有个单子叫包。我想知道里面是否只有一个特别的东西。

bag = ["drink"]
if only "drink" in bag:
    print 'There is only a drink in the bag'
else:
    print 'There is something else other than a drink in the bag'

当然，如果我在代码中加上“only”，那就错了。有什么简单的替代品吗？我试过几个类似的词。

Answer 1

使用内置的all()函数。

if bag and all(elem == "drink" for elem in bag):
    print("Only 'drink' is in the bag")

all()函数如下：

def all(iterable):
    for element in iterable:
        if not element:
            return False
    return True

因此，空列表将返回True。由于没有元素，它将完全跳过循环并返回True。因为是这种情况，您必须添加一个显式的and len(bag)或and bag，以确保包不是空的（()和[]是假的）。

此外，还可以使用set。

if set(bag) == {['drink']}:
    print("Only 'drink' is in the bag")

或者，类似地：

if len(set(bag)) == 1 and 'drink' in bag:
    print("Only 'drink' is in the bag")

所有这些都将使用列表中的0个或多个元素。