Numpy restrict array to given indices

Question

Let's say I have an array J and another array O of indices that I would like to restrict J to. If J was [1, 0, 9, 1] and O was [0, 3], something would happen to give me [1, 1]. Is there any numpy function for this?

Nguyễn Minh Hiếu · Accepted Answer · 2021-06-23 02:20:01Z

1

You can slice numpy array by selected indexes:

import numpy as np

arr = np.array([1, 0, 9, 1])
arr = arr[[0, 3]]  # >> array([1, 1])

answered Jun 23, 2021 at 2:20

Nguyễn Minh Hiếu

6771 gold badge5 silver badges12 bronze badges

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

Collectives™ on Stack Overflow

Numpy restrict array to given indices

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