Access the sole element of a set
I have a set
in Python from which I am removing elements one by one based on a condition. When the set is left with just 1 element, I need to return that element. How do I access this element from the set?
A simplified example:
S = set(range(5))
for i in range(4):
S = S - {i}
# now S has only 1 element: 4
return ? # how should I access this element
# a lame way is the following
# for e in S:
# return S
Use set.pop
:
>>> {1}.pop()
1
>>>
In your case, it would be:
return S.pop()
Note however that this will remove the item from the set. If this is undesirable, you can use min
|max
:
return min(S) # 'max' would also work here
Demo:
>>> S = {1}
>>> min(S)
1
>>> S
set([1])
>>> max(S)
1
>>> S
set([1])
>>>
I would use:
e = next(iter(S))
This is non-destructive and works even when there is more than one element in the set. Even better, it has an option to supply a default value e = next(iter(S), default)
.
You could also use unpacking:
[e] = S
The unpacking technique is likely to be the fastest way and it includes error checking to make sure the set has only one member. The downside is that it looks weird.
Sorry, late to the party. To access an element from a set you can always cast the set into a list and then you can use indexing to return the value you want.
In the case of your example:
return list(S)[0]