I would like to vectorize this NumPy operation:

```
for j in range(yt):
    for i in range(xt):
        y[j, i] = x[idx[j, i], j, i]

```

where `idx` contains axis-0 index to an `x` slice. Is there some simple way to do this?