How to get a 2D NumPy array with value 1 at indices represented by values in 1D NumPy array (Python)

Question

How to get a 2D np.array with value 1 at indices represented by values in 1D np.array in Python.

Example:

[1, 2, 5, 1, 2]

should be converted to

[[0, 1, 0, 0, 0, 0],
 [0, 0, 1, 0, 0, 0],
 [0, 0, 0, 0, 0, 1],
 [0, 1, 0, 0, 0, 0],
 [0, 0, 1, 0, 0, 0]]

Here you already know the width (shape[2]) value of the new array beforehand.

I can do it manually but is there any way to do it directly using NumPy methods for faster execution? The dimension of my array is quite large and I have to do this for all iteration. Thus, doing this manually for each iteration is quite computationally demanding.

Ch3steR · Accepted Answer · 2021-09-25 15:38:13Z

8

You can create a array with zeros using np.zeros. The shape the array should be (len(1D array), max(1D array)+1). Then use NumPy's indexing.

idx = [1, 2, 5, 1, 2]
shape = (len(idx), max(idx)+1)
out = np.zeros(shape)
out[np.arange(len(idx)), idx] = 1
print(out)
[[0. 1. 0. 0. 0. 0.]
 [0. 0. 1. 0. 0. 0.]
 [0. 0. 0. 0. 0. 1.]
 [0. 1. 0. 0. 0. 0.]
 [0. 0. 1. 0. 0. 0.]]

answered Sep 25, 2021 at 15:38

Ch3steR

20.8k4 gold badges34 silver badges66 bronze badges

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

Collectives™ on Stack Overflow

How to get a 2D NumPy array with value 1 at indices represented by values in 1D NumPy array (Python)

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related