I have an array A that has shape (480, 640, 3), and an array B with shape (480, 640).
How can I append these two as one array with shape (480, 640, 4)?
I tried np.append(A,B) but it doesn't keep the dimension, while the axis option causes the ValueError: all the input arrays must have same number of dimensions.
Use dstack:
>>> np.dstack((A, B)).shape
(480, 640, 4)
This handles the cases where the arrays have different numbers of dimensions and stacks the arrays along the third axis.
Otherwise, to use append or concatenate, you'll have to make B three dimensional yourself and specify the axis you want to join them on:
>>> np.append(A, np.atleast_3d(B), axis=2).shape
(480, 640, 4)
np.concatenate(A, B[...,None], axis=2)
TypeError: concatenate() takes at most 2 arguments (3 given) concatenate docs? They have priority over my old comments. In any case, I should have written np.concatenate([A, B[...,None]], axis=2). That is, put the arrays in a listconcatenate docs? They have priority over my old comments. In any case, I should have written np.concatenate([A, B[...,None]], axis=2). That is, put the arrays in a list
using np.stack should work
but the catch is both arrays should be of 2D form.
np.stack([A,B])
