How to extract the member from single-member set in python?

前端 未结 6 1296
感动是毒
感动是毒 2020-12-23 18:46

I recently encountered a scenario in which if a set only contained a single element, I wanted to do something with that element. To get the element, I settled on this appro

6条回答
  •  眼角桃花
    2020-12-23 19:11

    Tuple unpacking works.

    (element,) = myset
    

    (By the way, python-dev has explored but rejected the addition of myset.get() to return an arbitrary element from a set. Discussion here, Guido van Rossum answers 1 and 2.)

    My personal favorite for getting an arbitrary element is (when you have an unknown number, but also works if you have just one):

    element = next(iter(myset)) ¹
    

    1: in Python 2.5 and before, you have to use iter(myset).next()

提交回复
热议问题