python 定制类：索引与切片

__getitem__ 和 slice 切片

__getitem__ 的简单用法：

当一个类中定义了 __getitem__ 方法，那么它的实例对象便拥有了通过下标来索引的能力。

class A:
    def __getitem__(self, item):
        return item


a = A()
print(a[5], a[12])

5 12

用 __getitem__ 实现斐波那契数列：

class Fib:
    def __getitem__(self, item):
        if item == 0 or item == 1:
            return 1
        else:
            a, b = 1, 1
            for i in range(item - 1):
                a, b = b, a + b
            return b


fib = Fib()
for i in range(10):
    print(fib[i])

1
1
2
3
5
8
13
21
34
55

slice 切片

我们发现上个程序 fib[2] 这样的一个下标是没问题的，不过 fib[3:5] 这样的切片的话是会报错的。所以我们先了解下什么是切片。

a = slice(10)
a

slice(None, 10, None)

a = slice(1,10,2)
print(a.start)
print(a.stop)
print(a.step)
a

1
10
2





slice(1, 10, 2)

li = [1,2,3,4,5,6,7,8,9,0]
li[slice(2,6)]

[3, 4, 5, 6]

给斐波那契数列完成 slice 切片：

class Fib(object):
    def __getitem__(self, item):
        if isinstance(item, slice):
            print(item.start)
            print(item.stop)

            stop = item.stop
            if item.stop is None:
                stop = 100

            re_list = []

            a, b = 0, 1
            for i in range(stop):
                a, b = b, a + b
                re_list.append(a)

            return re_list[item.start:stop:item.step]

        elif item == 0 or item == 1:
            return 1
        else:
            a, b = 1, 1
            for i in range(item - 1):
                a, b = b, a + b
            return b


fib = Fib()
print(fib[1:20:2])

1
20
[1, 3, 8, 21, 55, 144, 377, 987, 2584, 6765]