🧩 Как соединить два массива Python NumPy: простая инструкция

import numpy as np

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

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

print(result)

[1 2 3 4 5 6]

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

В данной статье мы рассмотрим, как соединить два массива в библиотеке NumPy с помощью функции np.concatenate(). Мы также рассмотрим другие варианты объединения массивов.

1. Использование функции np.concatenate()

Функция np.concatenate() позволяет объединить один или более массивов по заданной оси. При этом размеры массивов должны совпадать по всем остальным осям.

import numpy as np

# Создание двух массивов
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

# Соединение массивов по оси 0
result = np.concatenate((arr1, arr2), axis=0)

print(result)

Результат выполнения кода:

[1 2 3 4 5 6]
    

2. Использование оператора np.vstack()

Оператор np.vstack() позволяет вертикально объединить массивы (соединить по оси 0).

import numpy as np

# Создание двух массивов
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

# Вертикальное объединение массивов
result = np.vstack((arr1, arr2))

print(result)

Результат выполнения кода:

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

3. Использование оператора np.hstack()

Оператор np.hstack() позволяет горизонтально объединить массивы (соединить по оси 1).

import numpy as np

# Создание двух массивов
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

# Горизонтальное объединение массивов
result = np.hstack((arr1, arr2))

print(result)

Результат выполнения кода:

[1 2 3 4 5 6]
    

4. Использование функции np.column_stack()

Функция np.column_stack() объединяет одномерные массивы в виде колонок двумерного массива.

import numpy as np

# Создание двух массивов
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

# Объединение массивов в виде колонок
result = np.column_stack((arr1, arr2))

print(result)

Результат выполнения кода:

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

5. Использование функции np.row_stack()

Функция np.row_stack() объединяет одномерные массивы в виде строк двумерного массива.

import numpy as np

# Создание двух массивов
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

# Объединение массивов в виде строк
result = np.row_stack((arr1, arr2))

print(result)

Результат выполнения кода:

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

6. Использование оператора np.dstack()

Оператор np.dstack() объединяет массивы вдоль третьей оси (оси глубины).

import numpy as np

# Создание двух массивов
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

# Объединение массивов вдоль третьей оси
result = np.dstack((arr1, arr2))

print(result)

Результат выполнения кода:

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

Вывод

В данной статье мы рассмотрели несколько способов соединения двух массивов в библиотеке NumPy. Команда np.concatenate() позволяет объединять массивы по заданной оси, операторы np.vstack() и np.hstack() позволяют вертикальное и горизонтальное объединение соответственно. Функции np.column_stack(), np.row_stack() и оператор np.dstack() предоставляют дополнительные возможности для объединения массивов. Выберите подходящий метод в зависимости от ваших потребностей и требуемого результата.