How to Convert 1-D Arrays as Columns into a 2-D Array in Python?

Arrays are fundamental data structures in programming, enabling us to store and manipulate collections of values efficiently. In this article, we will explore how to convert 1-D arrays into columns of a 2-D array using Python's NumPy library.

Understanding 1-D and 2-D Arrays

A 1-D array (one-dimensional array) represents a collection of elements arranged in a single row or column. Each element is accessed using an index indicating its position, such as [1, 2, 3, 4, 5].

A 2-D array (two-dimensional array) organizes elements in rows and columns, forming a grid or table structure where each element is identified by row and column indices ?

import numpy as np

# Example of a 2-D array
array_2d = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])
print(array_2d)

[[1 2 3]
 [4 5 6]
 [7 8 9]]

Using numpy.column_stack()

The np.column_stack() function takes multiple 1-D arrays and stacks them horizontally as columns to form a 2-D array ?

import numpy as np

# Input arrays
array1 = np.array([1, 2, 3, 4])
array2 = np.array([5, 6, 7, 8])

# Convert 1-D arrays into columns of a 2-D array
result = np.column_stack((array1, array2))
print(result)

[[1 5]
 [2 6]
 [3 7]
 [4 8]]

Multiple Arrays Example

import numpy as np

# Multiple 1-D arrays
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
arr3 = np.array([7, 8, 9])

# Stack as columns
result = np.column_stack((arr1, arr2, arr3))
print(result)

[[1 4 7]
 [2 5 8]
 [3 6 9]]

Using numpy.vstack() with Transpose

The np.vstack() function stacks arrays vertically. To convert 1-D arrays into columns, we stack them vertically and then transpose the result using .T ?

import numpy as np

# Input arrays
array1 = np.array([1, 2, 3, 4])
array2 = np.array([5, 6, 7, 8])
array3 = np.array([9, 10, 11, 12])

# Stack vertically then transpose
result = np.vstack((array1, array2, array3)).T
print(result)

[[ 1  5  9]
 [ 2  6 10]
 [ 3  7 11]
 [ 4  8 12]]

Using numpy.c_[] (Concatenate)

NumPy's c_[] provides a convenient way to concatenate arrays column-wise ?

import numpy as np

# Input arrays
x = np.array([1, 2, 3])
y = np.array([4, 5, 6])

# Concatenate as columns
result = np.c_[x, y]
print(result)

[[1 4]
 [2 5]
 [3 6]]

Comparison

Conclusion

NumPy provides multiple methods to convert 1-D arrays into columns of a 2-D array. Use column_stack() for clarity, vstack().T for vertical-first thinking, or c_[] for concise syntax. All methods are efficient and suitable for data analysis and scientific computing tasks.