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

Question

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

例如,如果我有一个N维ndarray arr,则arr[0]返回一个(N-1)维ndarray.

但是,如果我有1维ndarray x,x[0]则不返回0维ndarray,而是返回a numpy.int64,(如果x包含int64s).

最小的例子:

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, ...])