I have an array of probabilities:
JavaScript
x
2
1
l = [0.9, 0.2, 0.4]
2
and I want to make it 2d array:
JavaScript
1
4
1
l = [0.9 0.1
2
0.2 0.8
3
0.4 0.6]
4
What is the best way to do so?
Advertisement
Answer
One idea is use numpy.hstack
:
JavaScript
1
10
10
1
l = [0.9, 0.2, 0.4]
2
3
a = np.array(l)[:, None]
4
5
arr = np.hstack((a, 1 - a))
6
print (arr)
7
[[0.9 0.1]
8
[0.2 0.8]
9
[0.4 0.6]]
10
Or use numpy.c_
:
JavaScript
1
4
1
a = np.array(l)
2
arr = np.c_[a, 1 - a]
3
print (arr)
4