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

时间:2013-07-23 04:47:46

标签: python iterator set immutability pop

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

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

按预期返回:

'a'

换句话说,是否有任何方法可以从冻结集中“弹出”元素而不实际弹出它?

4 个答案:

答案 0 :(得分:10)

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

您的方法与任何方法一样好,但需要注意的是,从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']

答案 1 :(得分:10)

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

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

答案 2 :(得分:2)

你可以使用python 3:

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

答案 3 :(得分:0)

使用list

def get_frozenset_elements(frozen_set):
    frozenset_list = []
    for i in frozen_set:
        frozenset_list.append(i)
    return frozenset_list

list_ = get_frozenset_elements(frozenset("Benyamin"))

使用generator

def get_frozenset_elements(frozen_set):
    for i in frozen_set:
        yield i

gen_ = get_frozenset_elements(frozenset("Benyamin"))
next(gen_)
next(gen_)
next(gen_)
...