在numpy数组中得到第k个维度的第i个切片

Question

我有一个n维度numpy数组,我想获得-th维度的i第一个切片k.必须有比这更好的东西

# ... 
elif k == 5:
    b = a[:, :, :, :, :, i, ...]
# ...

Answer 1

b = a[(slice(None),) * k + (i,)]

手动构造索引元组.

如Python语言参考中所述,表单的表达式

a[:, :, :, :, :, i]

转换为

a[(slice(None), slice(None), slice(None), slice(None), slice(None), i)]

我们可以通过直接构建该元组而不是使用切片符号来实现相同的效果.(有一个小小的警告,即构建元组直接生成a[(i,)]而不是a[i]for k=0,但是NumPy会为标量处理这些内容i.)