从python中的（2，MN）矩阵读取（M，N）图像的值

Question

假设我有一个 (M,N) 图像 J（形状为 (M,N,3)）。我有一个 (2,MN) 矩阵 K 像这样：0 0 0 ... 0 1 1 1 ... 1 ............. M M M ... M0 1 2 ... N 1 2 3 ... N ............. 1 2 3 ... N现在我将上面的矩阵乘以一个 2 x 2 的矩阵，得到一个与 K 大小相同的新矩阵 T。现在我想创建一个新图像，其中这个新图像中的像素 (r,s) 等于旧图像像素的 (R,G,B) 值，该像素位于 r*N+M 列中T。如果可能，我想以矢量化的方式执行此操作。我不想使用 for 循环（我已经知道如何使用 for 循环来做到这一点，但它非常慢）。事实上，我对这个问题的兴趣是因为我想以矢量化的方式应用单应变换。任何帮助表示赞赏。这是使我想要清楚的循环版本：for r in range(0,M):    for s in range(0,N):        x, y = T[:,r*N+s]        new_image[r,s] = J[x,y]