Duplicar posible:
Python: Retrieve items from a set¿Hay alguna manera de obtener un artículo de un conjunto en O (1) vez?
Considere el siguiente código:
>>> item1 = (1,)
>>> item2 = (2,)
>>> s = set([item1, item2])
>>> s
set([(2,), (1,)])
>>> new_item = (1,)
>>> new_item in s
True
>>> new_item == item1
True
>>> new_item is item1
False
Así new_item
está en s
porque equivale a uno de sus artículos, pero es un objeto diferente
Lo que quiero es obtener item1
de s
dado new_item
es en s
.
Una solución que he llegado con es sencillo, pero no es muy eficiente:
def get_item(s, new_item):
for item in s:
if item == new_item:
return item
>>> get_item(s, new_item) is new_item
False
>>> get_item(s, new_item) is item1
True
Otra solución parece más eficiente, pero en realidad no funciona:
def get_item_using_intersection1(s, new_item):
return set([new_item]).intersection(s).pop()
Tampoco éste:
def get_item_using_intersection2(s, new_item):
return s.intersection(set([new_item])).pop()
Porque la intersección funciona de manera indefinida:
>>> get_item_using_intersection1(s, new_item) is new_item
True
>>> get_item_using_intersection1(s, new_item) is item1
False
>>> get_item_using_intersection2(s, new_item) is new_item
True
>>> get_item_using_intersection2(s, new_item) is item1
False
Si esto es importante, estoy usando Python 2.7 x64 en Windows 7, pero necesito una solución multiplataforma.
Gracias a todos. Se me ocurrió la siguiente solución temporal:
class SearchableSet(set):
def find(self, item):
for e in self:
if e == item:
return e
que será sustituido en el futuro con la siguiente solución (que es muy incompleta en este momento):
class SearchableSet(object):
def __init__(self, iterable=None):
self.__data = {}
if iterable is not None:
for e in iterable:
self.__data[e] = e
def __iter__(self):
return iter(self.__data)
def __len__(self):
return len(self.__data)
def __sub__(self, other):
return SearchableSet(set(self).__sub__(set(other)))
def add(self, item):
if not item in self:
self.__data[item] = item
def find(self, item):
return self.__data.get(item)
Pero ... La "solución ineficiente" que surgió ya es lineal. – kennytm
Creo que quiere decir * constante * tiempo –
@KennyTM, gracias, he editado el título de mi pregunta. – utapyngo