Функция NumPy vstack() в Python: синтаксис и примеры

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

Вот синтаксис функции vstack():

numpy.vstack((a1,a2,...))

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

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

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

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

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

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

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

import numpy as np

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

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

Выход:

[[1 2] 
 [3 4]]

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

В следующем примере предпринимается попытка объединить элементы двух одномерных массивов разной длины, что приводит к ошибке:

import numpy as np

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

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

Ошибка:

ValueError: all the input array dimensions for the concatenation axis must match exactly, but along dimension 1, the array at index 0 has size 2 and the array at index 1 has size 3

2) Функция vstack() в объединении двумерных массивов

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

import numpy as np

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

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

Выход:

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