Comparer 2 tableaux en utilisant numpy et l'attribution des valeurs à une troisième rangée

Question

Je

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

Je cherche un C qui ne

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

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

Comment puis-je faire cela?

Editer: J'ai commencé python aujourd'hui. Je suis donc tôt débutant en Python

Answer 1

np.where est faite à cet effet

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

Answer 2

Pour ce cas précis, vous pouvez simplement faire

(A > B).astype(int) 

Comme booléens convertir en int comme 0 et 1