交换二维数组中的两个元素

Question

我有一个形状为 (10296, 6) 的数组。我想交换子数组中的最后两个元素。

a = [[1, 2, 3, 4, 5, 6][1, 2, 3, 4, 5, 6]...

这样每个数组的5和6就换成了：

a = [[1, 2, 3, 4, 6, 5][1, 2, 3, 4, 6, 5]...

Answer 1

在 numpy 中尝试 advanced slicing。阅读更多 here -

import numpy as np

a = np.array([[1, 2, 3, 4, 5, 6],
              [1, 2, 3, 4, 5, 6]])

a[:,[4, 5]] = a[:,[5, 4]]

array([[1, 2, 3, 4, 6, 5],
       [1, 2, 3, 4, 6, 5]])

swap two elements in 2d array