使用numpy比较2个数组并将值分配给第三个数组

Question

我有

A = numpy.array([[ 0.52241976,  0.50960677,  0.34597965]])
B = numpy.array([[0.5, 0.5, 0.5]])

我正在寻找能够做到的C

if (A > B):
C[i] = 1

Expected C: [[ 1, 1, 0]]

我该怎么做？

编辑：我今天已经启动了python。所以我是Python初学者

Answer 1

np.where就是出于此目的

C = numpy.where(A > B, 1, 0)

Answer 2

对于这种特殊情况你可以做到

(A > B).astype(int)

当布尔转换为int为0和1