Python3 How to get the rest array when given a whole array and a sub-array?

Question

I have an array and a sub-array, like following:

import numpy as np    
total_array = np.arange(10)
sub_array = np.array([0, 1, 2, 3])

I was wondering how to get the rest_array = total_array - sub_array = np.array([4, 5, 6, 7, 8, 9])?

Any simple ideas? Thanks in advance.

RandomGuy · Accepted Answer · 2021-05-19 08:09:05Z

1

You can simply use np.setdiff1d to get the result you want

import numpy as np    
total_array = np.arange(10)
sub_array = np.array([0, 1, 2, 3])

rest_array = np.setdiff1d(total_array, sub_array)
rest_array

array([4, 5, 6, 7, 8, 9])

edited May 19, 2021 at 8:09

RandomGuy

1,1979 silver badges17 bronze badges

answered May 19, 2021 at 7:23

Shubham Periwal

2,2582 gold badges10 silver badges29 bronze badges

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

Collectives™ on Stack Overflow

Python3 How to get the rest array when given a whole array and a sub-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