NumPy: Convert two 1-D arrays into a 2-D array

Last update on December 21 2024 08:31:21 (UTC/GMT +8 hours)

Depth-Wise Conversion of 1D to 2D

Write a NumPy program to convert (in sequence depth wise (along the third axis)) two 1-D arrays into a 2-D array.
Sample array: (10,20,30), (40,50,60)

Pictorial Presentation:

Sample Solution:

Python Code:

# Importing the NumPy library with an alias 'np'
import numpy as np

# Creating NumPy arrays 'a' and 'b' with vertical shapes
a = np.array([[10], [20], [30]])
b = np.array([[40], [50], [60]])

# Stacking arrays 'a' and 'b' along the third axis using np.dstack
c = np.dstack((a, b))

# Printing the resulting array 'c'
print(c)

Sample Output:

[[[10 40]]                                                             
                                                                       
 [[20 50]]                                                             
                                                                       
 [[30 60]]] 

Explanation:

‘a = np.array([[10],[20],[30]])’ creates a 2D array a with shape (3, 1).

‘b = np.array([[40],[50],[60]])’ This line creates another 2D array b with shape (3, 1).

c = np.dstack((a, b)): The np.dstack() function is used to stack the two arrays ‘a’ and ‘b’ depth-wise along the third axis. The resulting array ‘c’ has the shape (3, 1, 2), where the first depth layer contains the elements of a and the second depth layer contains the elements of ‘b’.

Python-Numpy Code Editor: