Tunring an array of probabilities, into a binary array

Question

I am trying to convert a NumPy array of probabilities that sum up to 1.0, into a binary array. Essentially, I want to turn the max probability in the array into 1, and the rest into 0s. Of course, I know how to do this using Python, but I was hoping that NumPy or SciPy already come up with a built-in function for making the operation more performant (we are speaking about a matrix containing millions of such arrays).

Show your python code?

Divakar
– Divakar

2018-05-19 08:27:50 +00:00
Commented May 19, 2018 at 8:27 — Divakar
– Divakar, Commented May 19, 2018 at 8:27
Please provide a small example to clarify.

fferri
– fferri

2018-05-19 08:33:20 +00:00
Commented May 19, 2018 at 8:33 — fferri
– fferri, Commented May 19, 2018 at 8:33

fferri · Accepted Answer · 2018-05-19 08:35:50Z

2

Compare items with the maximum value, and cast to int

>>> a=np.array([0.435,0.24,0.241,0.13,0.56])

>>> np.int8( a == a.max() )
array([0, 0, 0, 0, 1], dtype=int8)

answered May 19, 2018 at 8:35

fferri

19k8 gold badges57 silver badges110 bronze badges

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

Collectives™ on Stack Overflow

Tunring an array of probabilities, into a binary 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