numpy: auspacken Array entlang einer Achse

Question

sagen, dass ich ein RGB-Bild habe:

from skimage import data 
img = data.astronaut() 
print(img.shape) # (512, 512, 3) 

Gibt es einen prägnanten numpy Befehl es entlang der Farbkanäle zu entpacken:

R, G, B = np.unpack(img, 2) # ? 

---

Was Ich mache gerade Verständnis verwendet:

R, G, B = (img[:, :, i] for i in range(3)) 

Aber gibt es kein einfacheres Kommando?

Answer 1

Alternativ Sie np.rollaxis verwenden -

R,G,B = np.rollaxis(img,2) 

Answer 2

Sie können die Länge-3 Dimension nach vorne transponieren und dann auspacken:

R, G, B = img.transpose((2, 0, 1)) 

Answer 3

Alternativ Sie np.split verwenden können:

R, G, B = np.split(img, img.shape[-1], axis=-1) 

Wenn Ihr Array von Form (Höhe, Breite, Kanal) ist, können Sie verwenden np.dsplit zu teilen entlang der Tiefe Dimension: