Python中多维数组的变量切片

Question

我有一个维数为（Nt，Nx0，Ny0）的数组。对于轴0中的每个索引，我想在轴1和轴2中随机选择一个“矩形”。矩形具有固定大小Nx1 x Nx2，其中Nx1<；Nx0和Ny1<；Ny0。你知道吗

此代码执行我要执行的操作：

import numpy as np

for i in range(Nt):
    x0ind = round(0.5*(Nx0-Nx1))+np.random.randint(-max_x, max_x)
    x1ind = x0ind+Nx1

    y0ind = round(0.5*(Ny0-Ny1))+np.random.randint(-max_y, max_y)
    y1ind = y0ind+Ny1

    ar1[i,:,:] = ar0[i,x0ind:x1ind,y0ind:y1ind]

我觉得使用numpy索引应该可以做到这一点，例如：

import numpy as np

Nt = 100
Nx0 = 70
Ny0 = 70

Nx1 = 50
Ny1 = 50

max_x = 5
max_y = 5

ar0 = np.random.rand(Nt, Nx0, Ny0)

x_ind = np.random.randint(-max_x, max_x)
y_ind = np.random.randint(-max_y, max_y)

ar1 = ar0[:, x_ind:x_ind+Nx1, y_ind:y_ind+Ny1]

然而，这是行不通的。在没有for循环的情况下可以这样做吗？你知道吗