比較2個數組numpy的和第三陣列

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