Compare two arrays in Numpy

Question

After comparison between an array total= np.full((3,3),[0, 1, 2]) and array a = np.array([1],[0],[1]), I am looking to get a new_array:

array([[0, 2],
       [1, 2],
       [0, 2]])

Please explain the relationship between the input and output. You've got two answers that take two different interpretations of what you want, and it's not clear whether either of them performs the operation you actually want. — user2357112
– user2357112, Commented May 25, 2019 at 19:32
Both of them produce the output you specified for this particular pair of inputs, but return np.array([[0, 2], [1, 2], [0, 2]]) would also do that. They don't behave the same way as each other on other outputs. — user2357112
– user2357112, Commented May 25, 2019 at 19:55

sentence · Accepted Answer · 2019-05-25 10:08:54Z

1

You can use list comprehension and np.delete:

import numpy as np

total= np.full((3,3),[0, 1, 2])
a = np.array([[1],[0],[1]])

new_array = np.array([np.delete(l, i) for l,i in zip(total,a)])

result

array([[0, 2],
       [1, 2],
       [0, 2]])

answered May 25, 2019 at 10:08

sentence

9,0414 gold badges36 silver badges41 bronze badges

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

Comments

Lante Dellarovere · Accepted Answer · 2019-05-25 19:29:19Z

0

Can apply a mask and reshape:

import numpy as np

total= np.full((3,3),[0, 1, 2])
a = np.array([[1],[0],[1]])

a = np.repeat(a, 3, axis=1)
new_total = total[total!=a].reshape(3, 2)

>>> print(new_total)
[[0 2]
 [1 2]
 [0 2]]

answered May 25, 2019 at 19:29

Lante Dellarovere

1,8582 gold badges9 silver badges10 bronze badges

Collectives™ on Stack Overflow

Compare two arrays in Numpy

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related