python - 如何将概率列表转换为二进制值

Question

我将 0 到 1 之间的概率列表存储在 NumPy 数组中。我如何隐藏这些值，以便任何值 >0.5 变为 1 而任何 <0.5 变为 0？

我所拥有的基本上是:

    model = [0.123,0.789,0.456,0.654]  

我想要的是:

    model = [0,1,0,1]

这只是将整个数组更改为 1s，我也试过了

我试过用循环来做:

for i in range(len(model)):
if i<0.5:
    model[i]=0
elif i>0.5:
    model[i]=1

这只是将整个数组更改为 1s，我也试过了

i = 0
while i <len(model):
    if model[i] < 0.5:
        model[i] = 0
        i + 1
    elif model[i] >0.5:
        model[i] = 1
        i + 1

但这也不起作用。

Answer 1

没有任何额外的库，你可以去...

model = [int(i > .5) for i in model]