I have an array arr and a list of indices that I want to get indices. I want to get subset of array corresponding to items in indices and the complement of that.
For example
for
arr = np.asarray([2, 4, 1, 5, 6])
indices = np.asarray([2, 4])
I would get
[1, 6] and [2, 4, 5]
Thanks
Using np.isin or np.in1d (using masks):
arr = np.asarray([2, 4, 1, 5, 6])
indices = np.asarray([2, 4])
m = np.in1d(np.arange(len(arr)), indices)
arr1, arr2 = arr[m], arr[~m]
arr1, arr2
>>array([1, 6]), array([2, 4, 5])
Alternatively, using np.setdiff1d for the complementary part (can be faster for larger arrays and indices):
arr1 = arr[indices]
arr2 = arr[np.setdiff1d(np.arange(len(arr)), indices, True)]
Collected from the Internet
Please contact [email protected] to delete if infringement.
Comments