从索引数组的函数返回等效的`：`

从索引数组的函数返回等效的`：`

NumPy有一个

np.s_[]

可用于构造

slice

和

Ellipsis

对象的助手：

def get_idx(k):    return np.s_[:] if k > 6 else np.s_[:k]    # or even np.s_[:None if k > 6 else k]

一般而言，

a[np.s_[ <stuff> ]]

与完全相同

a[ <stuff> ]

。