flip two values in rows of 2d numpy array

Question

I wrote the following code to flip two values in rows of 2d numpy array. However, I am wondering if there is a better way to do this. thanks in advance

def mtn(offg):
    for row in range(offg.shape[0]):
        point1 = random.randint(0, 139)
        point2 = random.randint(0, 139)
        temp = offg[row, point1]
        offg[row, point1] = offg[row, point2]
        offg[row, point2] = temp
    return offg

tstanisl · Accepted Answer · 2020-06-30 19:30:20Z

1

You could use advanced indexing:

offg[:,[p1, p2]] = offg[:,[p2, p1]]

answered Jun 30, 2020 at 19:30

tstanisl

14.3k3 gold badges31 silver badges45 bronze badges

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

Comments

Demetry Pascal · Accepted Answer · 2020-06-30 18:28:15Z

0

if points are same in each row, u can use code like

tmp = offg[:,p1].copy()
offg[:,p1] = offg[:,p2]
offg[:,p2] = tmp

answered Jun 30, 2020 at 18:28

Demetry Pascal

5671 gold badge9 silver badges20 bronze badges

3 Comments

MathPro Over a year ago

Thank you, unfortunately they are not the same

Demetry Pascal Over a year ago

So I don't see some problems in your way

MathPro Over a year ago

yes but I was looking for more advanced way to do it like the advanced indexing indicated by tstanisl above

Collectives™ on Stack Overflow

flip two values in rows of 2d numpy array

2 Answers 2

Comments

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related