python将一维概率数组转换为二维数组

【问题标题】：python transform 1d array of probabilities to 2d arraypython将一维概率数组转换为二维数组
【发布时间】：2021-07-04 05:12:16
【问题描述】：

我有一个概率数组：

l = [0.9, 0.2, 0.4]

我想把它做成二维数组：

l = [0.9 0.1
     0.2 0.8
     0.4 0.6]

最好的方法是什么？谢谢！

【问题讨论】：

【解决方案1】：

一个想法是使用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)

【讨论】：