在numpy数组的前n个维度中选择0

Question

我有一个形状为(1,2,3,4,5,6,...)(任意长度)的numpy数组，我想选择第一个n列中的第一个条目，即

def select_first_n_0(arr, n):
if n == 1:
return arr[0]
elif n == 2:
return arr[0][0]
elif n == 2:
return arr[0][0][0]
# ...

是否有更全面的表达?

Answer 1

可以使用递归

def select_first_n_0(arr, n):
if n == 1:
return arr[0]
return select_first_n_0(arr[0], n-1)

Answer 2

@hjpaul:

def select_first_n_0(arr, n):
return arr[tuple([0] * n)]