您可以使用python’s 自动
切片
slice：
>>> a = np.random.rand(3, 4, 5)>>> a[0, :, 0]array([ 0.48054702, 0.88728858, 0.83225113, 0.12491976])>>> a[(0, slice(None), 0)]array([ 0.48054702, 0.88728858, 0.83225113, 0.12491976])
该
slice方法显示为
slice(*start*, stop[, step])。如果仅传递一个参数，则将其解释为
slice(0, stop)。
在上面的例子中
:被转换成
slice(0, end)其等效于
slice(None)。
其他切片示例：
:5 -> slice(5)1:5 -> slice(1, 5)1: -> slice(1, None)1::2 -> slice(1, None, 2)
