Как объединить два массива в NumPy Python?

import numpy as np

array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])

combined_array = np.concatenate((array1, array2))

print(combined_array)
    

[1 2 3 4 5 6]
    

Как объединить два массива NumPy в Python

В программировании часто возникает необходимость объединить два массива в один, чтобы обрабатывать их как единое целое. В языке программирования Python с использованием библиотеки NumPy, эту задачу можно решить с помощью функции numpy.concatenate.

Прежде чем мы перейдем к объединению массивов, давайте вспомним, что такое массивы NumPy. NumPy - это библиотека Python, предоставляющая мощные инструменты для работы с многомерными массивами и матрицами. Она предоставляет высокопроизводительные структуры данных и операции над ними, что делает ее идеальной для работы с научными и численными данными.

Метод numpy.concatenate()

Функция numpy.concatenate() позволяет объединить два или более массивов вдоль определенной оси. Она принимает в качестве аргументов массивы, которые нужно объединить, и ось, вдоль которой нужно производить объединение.

import numpy as np

array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])

result = np.concatenate((array1, array2))
print(result)

В данном примере мы создаем два массива array1 и array2 с помощью функции np.array(). Затем мы используем функцию np.concatenate(), передавая в нее эти два массива в качестве аргументов.

Результатом будет объединенный массив [1, 2, 3, 4, 5, 6], который будет выведен на экран с помощью функции print().

Указание оси объединения

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

import numpy as np

array1 = np.array([[1, 2], [3, 4]])
array2 = np.array([[5, 6]])

result = np.concatenate((array1, array2), axis=0)
print(result)

В этом примере мы создаем два массива - двумерный массив array1 и одномерный массив array2. Затем, с помощью функции np.concatenate(), мы объединяем эти два массива вдоль горизонтальной оси (ось 0).

Результатом будет двумерный массив [[1, 2], [3, 4], [5, 6]], который будет выведен на экран с помощью функции print().

Объединение массивов разной формы

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

import numpy as np

array1 = np.array([[1, 2], [3, 4]])
array2 = np.array([5, 6])

result = np.concatenate((array1, array2), axis=0)
print(result)

В этом примере массивы array1 и array2 имеют разную форму, но так как оси по остальным измерениям совпадают (в данном случае 1-ое измерение), мы можем объединить их вдоль этой оси.

Результатом будет двумерный массив [[1, 2], [3, 4], [5, 6]], который будет выведен на экран.

Заключение

Функция numpy.concatenate() является мощным инструментом для объединения массивов в языке программирования Python с использованием библиотеки NumPy. Она позволяет объединять массивы вдоль заданной оси, при этом учитывая их форму. Этот инструмент особенно полезен при работе с многомерными данными и матрицами.