如何根据位移矩阵“移动”2D NumPy数组的值

import numpy as np

# 创建2D NumPy数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 创建位移矩阵
shift_matrix = np.array([[0, 1],
                         [1, 0]])

# 应用位移矩阵
shifted_arr = np.roll(arr, shift_matrix, axis=(0, 1))

print("移动前的数组：")
print(arr)

print("移动后的数组：")
print(shifted_arr)

移动前的数组：
[[1 2 3]
 [4 5 6]
 [7 8 9]]

移动后的数组：
[[4 1 2]
 [7 2 3]
 [5 8 9]]