Numpy Matrix Remove row that equals to an array

Question

I have a numpy matrix with 2 axis (row and columns) and an array. I want to remove the row in the matrix that equals to the array. For example, if the matrix is

[[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]

And the array is [1, 2, 3], then the output should be:

[[4, 5, 6],
[7, 8, 9]]

akuiper · Accepted Answer · 2021-09-25 17:27:15Z

3

Use:

a[~(a == b).all(1)]

Example:

a = np.arange(1, 10).reshape((3, 3))
b = np.arange(1, 4)

a[~(a == b).all(1)]
array([[4, 5, 6],
       [7, 8, 9]])

answered Sep 25, 2021 at 17:27

akuiper

216k33 gold badges363 silver badges380 bronze badges

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

Collectives™ on Stack Overflow

Numpy Matrix Remove row that equals to an array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related