我有一组概率:
l = [0.9, 0.2, 0.4]
我想把它变成二维数组:
l = [0.9 0.1
0.2 0.8
0.4 0.6]
这样做的最佳方法是什么? 谢谢!
答案 0 :(得分:2)
一个想法是使用 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]]
或者使用numpy.c_
:
a = np.array(l)
arr = np.c_[a, 1 - a]
print (arr)