【4.3】实现可切片的对象

#!/user/bin/env python
# -*- coding:utf-8 -*-

# 模式[start:end:step]
"""
    其中，第一个数字start表示切片开始位置，默认为0；
    第二个数字end表示切片截止（但不包含）位置（默认为列表长度）；
    第三个数字step表示切片的步长（默认为1）。
    当start为0时可以省略，当end为列表长度时可以省略，
    当step为1时可以省略，并且省略步长时可以同时省略最后一个冒号。
    另外，当step为负整数时，表示反向切片，这时start应该比end的值要大才行。
"""
aList = [3, 4, 5, 6, 7, 9, 11, 13, 15, 17]
# 返回包含原列表中所有元素的新列表
print(aList[::])
# 返回包含原列表中所有元素的逆序列表
print(aList[::-1])
# 隔一个取一个，获取偶数位置的元素
print(aList[::2])
# 隔一个取一个，获取奇数位置的元素
print(aList[1::2])
# 指定切片的开始和结束位置
print(aList[3:6])
# 切片结束位置大于列表长度时，从列表尾部截断
aList[0:100]
# 切片开始位置大于列表长度时，返回空列表
aList[100:]

# 在列表尾部增加元素
aList[len(aList):] = [9]
# 在列表头部插入元素
aList[:0] = [1, 2]
# 在列表中间位置插入元素
aList[3:3] = [4]
# 替换列表元素，等号两边的列表长度相等
aList[:3] = [1, 2]
# 等号两边的列表长度也可以不相等
aList[3:] = [4, 5, 6]
# 隔一个修改一个
aList[::2] = [0] * 3
print(aList)
# 隔一个修改一个
aList[::2] = ['a', 'b', 'c']
# 左侧切片不连续，等号两边列表长度必须相等
aList[::2] = [1,2]
# 删除列表中前3个元素
aList[:3] = []

# 切片元素连续
del aList[:3]
# 切片元素不连续，隔一个删一个
del aList[::2]

上述是切片常用的操作，下面代码将实现一个可切片的对象

#!/user/bin/env python
# -*- coding:utf-8 -*-
import numbers


class Group:
    """
    支持切片操作
    """
    def __init__(self, group_name, company_name, staffs):
        self.group_name = group_name
        self.company_name = company_name
        self.staffs = staffs

    def __reversed__(self):
        """
        倒序
        """
        self.staffs.reverse()

    def __getitem__(self, item):
        """
        实现这个函数之后，就可以进行切片操作
        """
        cls = type(self)
        if isinstance(item, slice):
            return cls(group_name=self.group_name, company_name=self.company_name, staffs=self.staffs[item])
        elif isinstance(item, numbers.Integral):
            return cls(group_name=self.group_name, company_name=self.company_name, staffs=[self.staffs[item]])

    def __len__(self):
        return len(self.staffs)

    def __iter__(self):
        """
        实现可迭代
        """
        return iter(self.staffs)

    def __contains__(self, item):
        """
        if 'zy' in group:
            print('yes')
        """
        if item in self.staffs:
            return True
        else:
            return False


staffs = ['zy1', 'zy2', 'zy3']
group = Group(group_name='user', company_name='imooc', staffs=staffs)
print(len(group))