Numpy：订阅多维数组中的最外层维度

Question

我正在处理多个多维数组。为简单起见，让我们考虑虚拟示例：

 array_list=[np.ones(3), np.ones((3,3,3)), np.ones((3,3)), np.ones(3)]

我需要订阅列表中每个数组的最外层维度。比如我的目标是根据最外层维度的指定范围将部分元素设置为零：

array_list[0][0:2]=0
array_list[1][:,:,0:2]=0
array_list[2][:,0:2]=0
array_list[3][0:2]=0

在我的实际应用程序中，我不知道我有多少个数组以及其中有多少个维度。

我想在 for 循环中完成任务：

for array in array_list:
    array[???]=0

但如果我不知道每个数组的维数，我正在努力实现这一点。

Answer 1

使用Ellipsis 选择除最后一个以外的所有维度（如果只有 1 个，则不选择任何内容）。

for array in array_list:
    array[..., 0:2] = 0