2D-Array an 3D-Array anhängen, die dritte Dimension erweitern

Question

Ich habe ein Array A mit der Form (480, 640, 3) und ein Array B mit der Form (480, 640).

Wie kann ich diese beiden als ein Array mit der Form (480, 640, 4) anhängen?

Ich versuchte , aber es hält nicht die Dimension, während die axis Option die ValueError: all the input arrays must have same number of dimensions verursacht.

Answer 1

Verwenden dstack:

>>> np.dstack((A, B)).shape 
(480, 640, 4) 

Diese behandelt die Fälle, in denen die Arrays eine unterschiedliche Anzahl von Dimensionen aufweisen und stapelt die Arrays entlang der dritten Achse.

Andernfalls verwenden append oder concatenate, werden Sie sich selbst machen müssen B dreidimensionale und geben Sie die Achse, die Sie wollen auf sie verbinden:

>>> np.append(A, np.atleast_3d(B), axis=2).shape 
(480, 640, 4)