从集合中选择单个项目：Python [英] select single item from a collection : Python

问题描述

我创建了一个实用程序函数来从生成器表达式返回预期的单个项目

I created a utility function to return the expected single item from an generator expression

print one(name for name in ('bob','fred') if name=='bob')

这是一个好方法去吧？

def one(g):
    try:
        val = g.next()
        try:
            g.next()
        except StopIteration:
            return val
        else:
            raise Exception('Too many values')
    except StopIteration:
        raise Exception('No values')

推荐答案

更简单的解决方案是使用元组解包。这已经做了你想要的一切，包括检查它是否包含一个项目。

A simpler solution is to use tuple unpacking. This will already do everything you want, including checking that it contains exactly one item.

单项：

 >>> name, = (name for name in ('bob','fred') if name=='bob')
 >>> name
 'bob'

物品太多：

>>> name, = (name for name in ('bob','bob') if name=='bob')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: too many values to unpack

没有项目：

>>> name, = (name for name in ('fred','joe') if name=='bob')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: need more than 0 values to unpack

这篇关于从集合中选择单个项目：Python的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！