Numpy mit wo mit verschiedenen Array-Größen

Question

So habe ich zwei PIL Bilder von RGBA. Was ich tun möchte, ist es, alle Orte zu finden, wo die RGB-Werte gleich sind und Alpha ist 255. Es sieht wie folgt aus:

from PIL import Image 
import numpy as np 
img1 = np.array(Image.open(/path/to/img1).convert('RGBA'), float).reshape(32,32,4) 
img2 = np.array(Image.open(/path/to/img2).convert('RGBA'), float).reshape(32,32,4) 

# Checks to see if RGB of img1 == RGB of img2 in all locations that A=255 
np.where((img1[:,:,:-1] == img2[:,:,:-1]) &\ # RGB check 
     (img1[:,:,3] == 255)) # Alpha check 

Aber dies führt zu operands could not be broadcast together with shapes (32,32,3) (32,32).
Ich dachte nicht, dass ich versuchte, sie zusammen zu übertragen, ich wollte nur die Indices finden, die ich wiederum in dieser Aussage ausstrahlt. Gibt es eine andere Möglichkeit, dies zu tun, oder eine Möglichkeit, ungleiche Formen nicht zu übertragen?

Answer 1

Verwenden .all(axis=-1) Orte zu finden, wo alle drei RGB-Werte gleich sind:

np.where((img1[..., :-1] == img2[..., :-1]).all(axis=-1) 
     & (img1[..., 3] == 255)) 

---

Als mgilson points out hat (img1[..., :-1] == img2[..., :-1])(32, 32, 3) zu formen. Der Aufruf .all(axis-1) reduziert die letzte Achse auf einen Skalar Booleschen Wert, so

(img1[..., :-1] == img2[..., :-1]).all(axis=-1) 

(32, 32) Form hat. Dies entspricht der Form (img1[..., 3] == 255), so dass diese booleschen Arrays mit bitwise-and operator, & kombiniert werden können.