【4.3】实现可切片的对象

 1 #!/user/bin/env python
 2 # -*- coding:utf-8 -*-
 3 
 4 # 模式[start:end:step]
 5 """
 6     其中，第一个数字start表示切片开始位置，默认为0；
 7     第二个数字end表示切片截止（但不包含）位置（默认为列表长度）；
 8     第三个数字step表示切片的步长（默认为1）。
 9     当start为0时可以省略，当end为列表长度时可以省略，
10     当step为1时可以省略，并且省略步长时可以同时省略最后一个冒号。
11     另外，当step为负整数时，表示反向切片，这时start应该比end的值要大才行。
12 """
13 aList = [3, 4, 5, 6, 7, 9, 11, 13, 15, 17]
14 # 返回包含原列表中所有元素的新列表
15 print(aList[::])
16 # 返回包含原列表中所有元素的逆序列表
17 print(aList[::-1])
18 # 隔一个取一个，获取偶数位置的元素
19 print(aList[::2])
20 # 隔一个取一个，获取奇数位置的元素
21 print(aList[1::2])
22 # 指定切片的开始和结束位置
23 print(aList[3:6])
24 # 切片结束位置大于列表长度时，从列表尾部截断
25 aList[0:100]
26 # 切片开始位置大于列表长度时，返回空列表
27 aList[100:]
28 
29 # 在列表尾部增加元素
30 aList[len(aList):] = [9]
31 # 在列表头部插入元素
32 aList[:0] = [1, 2]
33 # 在列表中间位置插入元素
34 aList[3:3] = [4]
35 # 替换列表元素，等号两边的列表长度相等
36 aList[:3] = [1, 2]
37 # 等号两边的列表长度也可以不相等
38 aList[3:] = [4, 5, 6]
39 # 隔一个修改一个
40 aList[::2] = [0] * 3
41 print(aList)
42 # 隔一个修改一个
43 aList[::2] = ['a', 'b', 'c']
44 # 左侧切片不连续，等号两边列表长度必须相等
45 aList[::2] = [1,2]
46 # 删除列表中前3个元素
47 aList[:3] = []
48 
49 # 切片元素连续
50 del aList[:3]
51 # 切片元素不连续，隔一个删一个
52 del aList[::2]

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

 1 #!/user/bin/env python
 2 # -*- coding:utf-8 -*-
 3 import numbers
 4 
 5 
 6 class Group:
 7     """
 8     支持切片操作
 9     """
10     def __init__(self, group_name, company_name, staffs):
11         self.group_name = group_name
12         self.company_name = company_name
13         self.staffs = staffs
14 
15     def __reversed__(self):
16         """
17         倒序
18         """
19         self.staffs.reverse()
20 
21     def __getitem__(self, item):
22         """
23         实现这个函数之后，就可以进行切片操作
24         """
25         cls = type(self)
26         if isinstance(item, slice):
27             return cls(group_name=self.group_name, company_name=self.company_name, staffs=self.staffs[item])
28         elif isinstance(item, numbers.Integral):
29             return cls(group_name=self.group_name, company_name=self.company_name, staffs=[self.staffs[item]])
30 
31     def __len__(self):
32         return len(self.staffs)
33 
34     def __iter__(self):
35         """
36         实现可迭代
37         """
38         return iter(self.staffs)
39 
40     def __contains__(self, item):
41         """
42         if 'zy' in group:
43             print('yes')
44         """
45         if item in self.staffs:
46             return True
47         else:
48             return False
49 
50 
51 staffs = ['zy1', 'zy2', 'zy3']
52 group = Group(group_name='user', company_name='imooc', staffs=staffs)
53 print(len(group))