在python中交换两行

Question

给定m和n，交换下面给出的2-D NumPy数组的第m行和第n行。

import numpy as np
a = np.array([[4, 3, 1],
[5, 7, 0],
[9, 9, 3],
[8, 2, 4]])

示例：

输入1:

0
2

输出1:

array([[9, 9, 3],
[5, 7, 0],
[4, 3, 1],
[8, 2, 4]])

Answer 1

您可以使用NumPy高级索引：

a[(m, n), :] = a[(n, m), :]