numpy.hstack() in Python

numpy.hstack() function stacks arrays in sequence horizontally (column-wise). It joins arrays along their second axis for 2D arrays or flattens and joins them for 1D arrays. This is useful for combining arrays side by side.

Example:

import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

res = np.hstack((a, b))
print(res)

import numpy as np

a = np.array([1, 2, 3])

b = np.array([4, 5, 6])

​

res = np.hstack((a, b))

print(res)

[1 2 3 4 5 6]

Arrays a and b are horizontally stacked to form one combined 1D array.

Syntax

numpy.hstack(tup, *, dtype=None, casting='same_kind')

Parameters:

Returns: This function returns horizontally stacked array of the input arrays.

Examples

Example 1: Horizontal stacking of 2D arrays

import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

res = np.hstack((a, b))
print(res)

import numpy as np

a = np.array([[1, 2], [3, 4]])

b = np.array([[5, 6], [7, 8]])

​

res = np.hstack((a, b))

print(res)

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

Each row of a and b is concatenated horizontally to form a wider 2D array.

Example 2: Stacking arrays of different shapes (raises an error)

import numpy as np
a = np.array([[1, 2]])
b = np.array([[3, 4], [5, 6]])

res = np.hstack((a, b))

import numpy as np

a = np.array([[1, 2]])

b = np.array([[3, 4], [5, 6]])

​

res = np.hstack((a, b))

Output

ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 0, the array a...

Arrays must be compatible in shape except along the concatenation axis.

Example 3: Horizontal stacking with negative numbers

import numpy as np
a = np.array([-1, -2, -3])
b = np.array([4, 5, 6])

res = np.hstack((a, b))
print(res)

import numpy as np

a = np.array([-1, -2, -3])

b = np.array([4, 5, 6])

​

res = np.hstack((a, b))

print(res)

[-1 -2 -3  4  5  6]

Works with negative integers too. The resulting array preserves the sign of the original numbers.