Python 选择所有内容的“：”的替代项

我知道这一点

np.arange(3)[:]

：

意味着我需要来自特定维度的所有元素。现在，我需要在我想要的元素方面保持灵活性。我的功能类似于：

def pickSomething(index):
    return np.arange(3)[index]

其中

索引可以是0、1或2。但是，我也希望能够返回整个数组而不转换它
pickSomething（：）
自然是无效语法。如果我选择了某个（无）

，我会创建一个我不想创建的额外维度

---

tl；dr：除了

：

之外，还有其他方法吗？我可以在

pickSomething（index）

这样的函数中传递，这将生成未更改的

np.arange（3）

数组？

我想您正在寻找

省略号

对象

>>> import numpy as np
>>> def pickSomething(index):
...     return np.arange(3)[index]
>>> pickSomething(Ellipsis)
array([0, 1, 2])

---

在Python3中，

…

是

省略号的简写形式

看看这是否有用-。@Divakar确实有用。