如何从numpy中的1维数组中提取零维切片

Question

有没有办法从一维数组切割零维子阵列？

例如，如果我有一个N维的ndarray arr ， arr[0]返回一个（N-1）维的ndarray 。

但是，如果我有一维ndarray x ， x[0]不返回0维ndarray，而是返回numpy.int64 ，（如果x包含int64 s）。

最小的例子：

def increment(zero_d_array):
    zero_d_array[...] = zero_d_array + 1

counter = numpy.array(0)  # a zero-dimensional array containing scalar 0
increment(counter)        # success; counter is now 1

counters = numpy.zeros(3, dtype=int)  # [0, 0, 0]
increment(counter[1])    # fails; counter[1] is a numpy.int64, not a 0-D array

我意识到上面的方法可以用increment(counter[1:2]) ，但只是因为increment()恰好适用于0-D和1-D输入。 并非所有功能都会如此灵活。

Answer 1

使用省略号：

increment(counter[1, ...])