python中列表切片语法的问题

Question

python 的文档中提到了扩展索引语法。

slice([start], stop[, step])

使用扩展索引语法时也会生成切片对象。例如：a[start:stop:step] 或 a[start:stop, i]。请参阅 itertools.islice() 了解返回迭代器的替代版本。

a[start:stop:step] 按描述工作。但第二个呢？它是如何使用的？

Answer 1

a[start:stop,i] 调用方法 a.__getitem__((slice(start,stop,None), i))。

如果 a 是列表，则会引发 TypeError，但如果 a 是 numpy 数组，则它是有效且有用的表示法。事实上，我相信 Numpy 的开发者要求 Python 的开发者精确地扩展有效的 Python 切片表示法，以便 numpy 数组切片表示法可以更容易地实现。

例如，

import numpy as np
arr=np.arange(12).reshape(4,3)
print(arr)
# [[ 0  1  2]
#  [ 3  4  5]
#  [ 6  7  8]
#  [ 9 10 11]]

1:3 选择第 1 行和第 2 行，2 选择第三列：

print(arr[1:3,2])
# [5 8]

PS。要试验哪个切片被发送到 __getitem__，您可以
玩一下这个玩具代码：

class Foo(list):
    def __getitem__(self,key):
        return repr(key)

foo=Foo(range(10))
print(foo[1:5,1,2])
# (slice(1, 5, None), 1, 2)

Answer 2

符号 [:,:] 用于切片 多维数组。 Python 默认情况下没有任何多维数组，但语法支持它，例如 numpy 利用这个语法。