重複の可能性:
Python:セットからアイテムを取得する
次のコードを検討してください。
>>> 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
それはそのアイテムの1つと同等new_item
でs
あるためですが、それは別のオブジェクトです。
私が欲しいのは与えられたものitem1
から得ることです。s
new_item
s
私が思いついた解決策の1つは簡単ですが、あまり効率的ではありません。
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
別の解決策はより効率的であるように見えますが、実際には機能しません。
def get_item_using_intersection1(s, new_item):
return set([new_item]).intersection(s).pop()
これも:
def get_item_using_intersection2(s, new_item):
return s.intersection(set([new_item])).pop()
交差点は未定義の方法で機能するため:
>>> 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
これが問題になる場合は、Windows7でPython2.7 x64を使用していますが、クロスプラットフォームソリューションが必要です。
みんなありがとう。私は次の一時的な解決策を思いついた:
class SearchableSet(set):
def find(self, item):
for e in self:
if e == item:
return e
これは将来、次のソリューションに置き換えられます(現在は非常に不完全です)。
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)