mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-26 18:29:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

SF patch #678899: Save time and memory by using itertools in sets module.

Browse source
This commit is contained in:
Raymond Hettinger 2003-02-02 14:27:19 +00:00
parent efb9625b81
commit a3a53180c0
1 changed files with 11 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

30
Lib/sets.py
View file

@ -57,7 +57,7 @@ what's tested is actually `z in y'.
__all__ = ['BaseSet', 'Set', 'ImmutableSet'] __all__ = ['BaseSet', 'Set', 'ImmutableSet']
from itertools import ifilter
class BaseSet(object): class BaseSet(object):
"""Common base class for mutable and immutable sets.""" """Common base class for mutable and immutable sets."""
@ -182,7 +182,7 @@ class BaseSet(object):
little, big = self, other little, big = self, other
else: else:
little, big = other, self little, big = other, self
common = filter(big._data.has_key, little._data) common = ifilter(big._data.has_key, little)
return self.__class__(common) return self.__class__(common)
def intersection(self, other): def intersection(self, other):
@ -204,12 +204,10 @@ class BaseSet(object):
value = True value = True
selfdata = self._data selfdata = self._data
otherdata = other._data otherdata = other._data
for elt in selfdata: for elt in ifilter(otherdata.has_key, selfdata, True):
if elt not in otherdata: data[elt] = value
data[elt] = value for elt in ifilter(selfdata.has_key, otherdata, True):
for elt in otherdata: data[elt] = value
if elt not in selfdata:
data[elt] = value
return result return result
def symmetric_difference(self, other): def symmetric_difference(self, other):
@ -228,11 +226,9 @@ class BaseSet(object):
return NotImplemented return NotImplemented
result = self.__class__() result = self.__class__()
data = result._data data = result._data
otherdata = other._data
value = True value = True
for elt in self: for elt in ifilter(other._data.has_key, self, True):
if elt not in otherdata: data[elt] = value
data[elt] = value
return result return result
def difference(self, other): def difference(self, other):
@ -264,10 +260,8 @@ class BaseSet(object):
self._binary_sanity_check(other) self._binary_sanity_check(other)
if len(self) > len(other): # Fast check for obvious cases if len(self) > len(other): # Fast check for obvious cases
return False return False
otherdata = other._data for elt in ifilter(other._data.has_key, self, True):
for elt in self: return False
if elt not in otherdata:
return False
return True return True
def issuperset(self, other): def issuperset(self, other):
@ -275,9 +269,7 @@ class BaseSet(object):
self._binary_sanity_check(other) self._binary_sanity_check(other)
if len(self) < len(other): # Fast check for obvious cases if len(self) < len(other): # Fast check for obvious cases
return False return False
selfdata = self._data for elt in ifilter(self._data.has_key, other, True):
for elt in other:
if elt not in selfdata:
return False return False
return True return True