Функция column_stack

Функция column_stack объединяет одномерные массивы как столбцы, формируя двумерный массив. Если входные массивы уже двумерные, функция работает аналогично hstack. Первым параметром передаётся кортеж или список массивов для объединения.

Синтаксис

np.column_stack(tup)

Пример

Давайте объединим два одномерных массива в двумерный массив, где каждый из них станет столбцом:

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
res = np.column_stack((arr1, arr2))
print(res)

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

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

Пример

При объединении трёх одномерных массивов они становятся тремя столбцами:

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.array([7, 8, 9])
res = np.column_stack((a, b, c))
print(res)

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

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

Пример

Если передать двумерные массивы, column_stack объединит их по столбцам, как и hstack:

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
res = np.column_stack((arr1, arr2))
print(res)

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

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

Пример

Одномерный массив и скаляр можно объединить, при этом скаляр повторяется для каждой строки:

import numpy as np

arr = np.array([1, 2, 3])
res = np.column_stack((arr, 0))
print(res)

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

[[1 0]
 [2 0]
 [3 0]]

Смотрите также

функцию vstack,
которая объединяет массивы по строкам
функцию hstack,
которая объединяет массивы по горизонтали
функцию stack,
которая объединяет массивы по новой оси
функцию concatenate,
которая объединяет массивы по существующей оси