python transform 1d array of probabilities to 2d array

Question

I have an array of probabilities:

l = [0.9, 0.2, 0.4]

and I want to make it 2d array:

l = [0.9 0.1
     0.2 0.8
     0.4 0.6]

What is the best way to do so?

jezrael · Accepted Answer · 2021-04-08 10:43:15Z

2

One idea is use numpy.hstack:

l = [0.9, 0.2, 0.4]

a = np.array(l)[:, None]

arr = np.hstack((a, 1 - a))
print (arr)
[[0.9 0.1]
 [0.2 0.8]
 [0.4 0.6]]

a = np.array(l)
arr = np.c_[a, 1 - a]
print (arr)

answered Apr 8, 2021 at 10:36

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

1 Answer 1