python-集合操作的重写

'''
练习: 实现有序集合类OrderSet(),能实现两个集合的交集 &,并集 |,补集 -, 对称补集: ^
==,!= , 等操作
要求: 集合内部list存储
'''
class OrderSet():
    def __init__(self,iterable):
        self.data = [x for x in iterable]
    def __repr__(self):
        return "OrderSet({})".format(self.data)
    def __and__(self,other):
        print("__and__ are called")
        lst = []
        for i in self.data:
            for j in other.data:
                if i == j:
                    lst.append(i)
                    break
        return OrderSet(lst)
    def __or__(self,other):
        print("__or__ are called")
        s1 = set(self.data)
        s2 = set(other.data)
        s3 = s1 | s2
        return OrderSet(list(s3))
    def __xor__(self,other):
        s1 = set(self.data)
        s2 = set(other.data)
        s3 = s1 ^ s2
        return OrderSet(s3)
    def __eq__(self,other):
        print("__eq__ are called")
        for index in range(len(self.data)):
            if self.data[index] != other.data[index]:
                return False
        return True
    def __ne__(self,other):
        print("__ne__ are called")
        for index in range(len(self.data)):
            if self.data[index] == other.data[index]:
                return False
        return True

lst1 = OrderSet([1,2,3,4])
lst2 = OrderSet([3,4,5,6])
lst3 = OrderSet([1,2,3,4])
print(lst1 & lst2)
print(lst1 | lst2)
print(lst1 ^ lst2)
print(lst1 == lst3)
print(lst1 != lst2)
print(lst1 != lst3)

 

你可能感兴趣的:(Python)