python 上的一列到两列列表

Question

早上好。我有清单，其中包含 2 个概率：

[[0.84285885 0.15714115]
 [0.67180324 0.32819676]
 [0.95325917 0.04674083]
 ...
 [0.38825085 0.61174915]
 [0.45577896 0.54422104]
 [0.7975929  0.2024071 ]]

和另一个列表，仅包含一个：

[[5.4913871e-02]
 [7.6471776e-02]
 [4.1644130e-04]
 ...
 [9.6865553e-01]
 [9.8211896e-01]
 [3.5812762e-01]]

如何将此列表转换为第一个列表的形式？将 1-x 应用于每个元素并获得相同形状的列表。谢谢

Answer 1

如果你有 numpy 数组：

a = np.array(
    [
        [5.4913871e-02],
        [7.6471776e-02],
        [4.1644130e-04],
        [9.6865553e-01],
        [9.8211896e-01],
        [3.5812762e-01],
    ]
)

然后：

print(np.column_stack((a[:, 0], 1 - a[:, 0])))

打印：

[[5.49138710e-02 9.45086129e-01]
 [7.64717760e-02 9.23528224e-01]
 [4.16441300e-04 9.99583559e-01]
 [9.68655530e-01 3.13444700e-02]
 [9.82118960e-01 1.78810400e-02]
 [3.58127620e-01 6.41872380e-01]]

python 上的一列到两列列表

One column to two column list on python

python

list

calculated-columns