如何从冻结集中获取任意元素？

Question

我想从冻结集中获取一个元素(当然,不修改它,因为frozensets是不可变的).到目前为止我找到的最佳解决方案是:

s = frozenset(['a'])
iter(s).next()

按预期返回:

'a'

换句话说,有没有任何方法可以从冻结集中"弹出"一个元素而不实际弹出它？

Answer 1

(总结评论中给出的答案)

您的方法与任何方法一样好,但需要注意的是,从Python 2.6开始,您应该使用next(iter(s))而不是iter(s).next().

如果您想要一个随机元素而不是任意元素,请使用以下内容:

import random
random.sample(s, 1)[0]

以下是一些示例,展示了这两者之间的区别:

>>> s = frozenset("kapow")
>>> [next(iter(s)) for _ in range(10)]
['a', 'a', 'a', 'a', 'a', 'a', 'a', 'a', 'a', 'a']

>>> import random
>>> [random.sample(s, 1)[0] for _ in range(10)]
['w', 'a', 'o', 'o', 'w', 'o', 'k', 'k', 'p', 'k']

Answer 2

如果您知道冻结集中只有一个元素,则可以使用可迭代解包:

s = frozenset(['a'])
x, = s

这有点是原始问题的一个特例,但它有时会派上用场.

如果你有很多这样的事情可能会比下一个更快(iter ..:

>>> timeit.timeit('a,b = foo', setup='foo = frozenset(range(2))', number=100000000)
5.054765939712524
>>> timeit.timeit('a = next(iter(foo))', setup='foo = frozenset(range(2))', number=100000000)
11.258678197860718

Answer 3

您可以使用python 3：

>>> s = frozenset(['a', 'b', 'c', 'd'])
>>> x, *_ = s
>>> x
'a'
>>> _, x, *_ = s
>>> x
'b'
>>> *_, x, _ = s
>>> x
'c'
>>> *_, x = s
>>> x
'd'