python将一维概率数组转换为二维数组
Posted
技术标签:
【中文标题】python将一维概率数组转换为二维数组【英文标题】:python transform 1d array of probabilities to 2d array 【发布时间】: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)
【讨论】:
以上是关于python将一维概率数组转换为二维数组的主要内容,如果未能解决你的问题,请参考以下文章