Функция NumPy concatenate() в Python с примерами

Функция NumPy concatenate() в Python позволяет объединить два или более массивов в один массив.

Вот базовый синтаксис функции np.concatenate():

np.concatenate((a1,a2,...),axis=0)

В этом синтаксисе функция concatenate() объединяет элементы последовательности массивов (a1, a2, …) в один массив. Массивы в последовательности должны иметь одинаковую форму.

axis определяет ось, вдоль которой функция будет объединять массивы. Если ось — None, функция выровняет массивы перед объединением.

Функция concatenate() возвращает объединенный массив.

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

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

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

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

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

import numpy as np

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

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

Выход:

[1 2 3 4]

В этом примере функция concatenate() объединяет элементы массива a и b в один массив c.

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

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

import numpy as np

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

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

Выход:

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

Вывод показывает, что функция concatenate() объединяет два массива по вертикали, поскольку по умолчанию аргумент axis равен нулю.

Если ось одна, функция concatenate() объединит два массива по горизонтали. Например:

import numpy as np

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

c = np.concatenate((a, b), axis=1)
print(c)

Выход:

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