根据另一个np.array python的值设置np.array的值

Question

我有一个numpy数组A，它包含0到1之间的值。我想创建另一个numpy数组y，如果A（i）＆gt; = 0.5，则y（i）的值为1，y （i）如果A（i）<0则= 0。 0.5。我使用了以下python代码：

ForkJoinWorkerThreadFactory

有没有办法在一行命令而不是三行中执行此功能？

Answer 1

import numpy

A = numpy.random.rand(10)
print(A)

数组 A ：

[ 0.76702953  0.89697124  0.54573644  0.48079479  0.39556016  0.50646642
  0.45998033  0.11159339  0.69824144  0.37451713]

创建另一个numpy数组y，如果y(i) = 1则为A(i) >= 0.5，y(i) = 0为A(i) < 0.5。

Y = (A >= 0.5).astype(int)
print(Y)

数组 Y ：

[1 1 1 0 0 1 0 0 1 0]