Функция NumPy hstack() в Python: примеры использования

• Что такое функция NumPy hstack() в Python?
• Примеры функции NumPy hstack()
• 1) Использование функции numpy hstack() для объединения элементов одномерных массивов
• 2) Объединение элементов двумерных массивов

Что такое функция NumPy hstack() в Python?

Функция hstack() в Python объединяет элементы двух или более массивов в один массив по горизонтали (по столбцам).

Ниже показан синтаксис функции hstack():

numpy.hstack((a1,a2,...))

В этом синтаксисе (a1, a2, …) представляет собой последовательность массивов типа ndarray.

Все массивы a1, a2, .. должны иметь одинаковую форму по всем осям, кроме второй. Если все массивы являются одномерными, то они могут иметь любую длину.

Если вы хотите объединить два или более массивов по вертикали, вы можете использовать функцию vstack().

Примеры функции NumPy hstack()

Давайте рассмотрим несколько примеров использования функции hstack().

1) Использование функции numpy hstack() для объединения элементов одномерных массивов

В следующем примере функция hstack() используется для объединения двух одномерных массивов по горизонтали:

import numpy as np

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

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

Выход:

[1 2 3 4 5]

Обратите внимание, что для одномерных массивов входные массивы могут иметь разную длину, как показано в примере выше.

2) Объединение элементов двумерных массивов

В следующем примере функция hstack() используется для объединения элементов двух двумерных массивов:

import numpy as np

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

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

Выход:

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