swap two elements in 2d array

Question

I have an array of the shape (10296, 6). I want to swap the two last elements in the subarray.

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

So that 5 and 6 of each array is swapped into:

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

Akshay Sehgal · Accepted Answer · 2021-10-11 16:36:23Z

3

Try advanced slicing in numpy. Read more 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]])

answered Oct 11, 2021 at 16:36

Akshay Sehgal

19.4k3 gold badges26 silver badges57 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

oh lol, i knew i was off

hey, if it works, its fine :)