Append 2D array to 3D array, extending third dimension

by

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)

More Related Contents:

  1. Is there a NumPy function to return the first index of something in an array?
  2. Understanding NumPy’s einsum
  3. Intuition and idea behind reshaping 4D array to 2D array in NumPy
  4. Is there any numpy group by function?
  5. Removing nan values from an array
  6. Why is numpy’s einsum faster than numpy’s built in functions?
  7. How to return 0 with divide by zero
  8. Reshape an array in NumPy
  9. Most efficient way to forward-fill NaN values in numpy array
  10. NumPy array initialization (fill with identical values)
  11. Why does numpy.zeros takes up little space
  12. Generalise slicing operation in a NumPy array
  13. How to split a numpy array in fixed size chunks with and without overlap?
  14. Parameters to numpy’s fromfunction
  15. List to array conversion to use ravel() function
  16. Efficient distance calculation between N points and a reference in numpy/scipy
  17. Fastest save and load options for a numpy array
  18. Numpy find indices of groups with same value
  19. numpy with python: convert 3d array to 2d
  20. How is numpy’s fancy indexing implemented?
  21. Convert row vector to column vector in NumPy
  22. How can I slice each element of a numpy array of strings?
  23. numpy division with RuntimeWarning: invalid value encountered in double_scalars
  24. How does the axis parameter from NumPy work?
  25. Sorting a python array/recarray by column
  26. Numpy Array Get row index searching by a row
  27. Resample a numpy array
  28. Creating a numpy array of 3D coordinates from three 1D arrays
  29. How to use Python multiprocessing Pool.map to fill numpy array in a for loop
  30. Is this the best way to add an extra dimension to a numpy array in one line of code?

Leave a Comment