Как добавить столбец в массив NumPy (с примерами)

Вы можете использовать один из следующих методов, чтобы добавить столбец в массив NumPy:

Способ 1: добавить столбец в конец массива

np.append(my_array, [[value1], [value2], [value3], ...], axis= 1 )

Способ 2: вставить столбец в определенную позицию массива

np.insert (my_array, 3 , [value1, value2, value3, ...], axis= 1 ) 

В следующих примерах показано, как использовать каждый метод на практике.

Пример 1: добавить столбец в конец массива NumPy

Предположим, у нас есть следующий массив NumPy:

import numpy as np

#create NumPy array
my_array = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])

#view NumPy array
my_array

array([[1, 2, 3, 4],
 [5, 6, 7, 8]])

Мы можем использовать следующий синтаксис, чтобы добавить столбец в конец массива NumPy:

#append column to end of NumPy array
new_array = np.append(my_array, [[10], [13]], axis= 1 )

#view updated array
new_array

array([[ 1, 2, 3, 4, 10],
 [ 5, 6, 7, 8, 13]])

Пример 2: вставка столбца в определенную позицию массива NumPy

Предположим, у нас есть следующий массив NumPy:

import numpy as np

#create NumPy array
my_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

#view NumPy array
my_array

array([[1, 2, 3],
 [4, 5, 6],
 [7, 8, 9]])

Мы можем использовать следующий синтаксис, чтобы вставить новый столбец перед столбцом в индексной позиции 2 массива NumPy:

#insert new column before column in index position 2
new_array = np.insert (my_array, 2 , [10, 13, 19], axis= 1 )

#view updated array
new_array

array([[ 1, 2, 10, 3],
 [ 4, 5, 13, 6],
 [ 7, 8, 19, 9]])

Обратите внимание, что новый столбец значений был вставлен перед столбцом в индексной позиции 2.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные операции в NumPy:

Как добавить массив Numpy в фрейм данных Pandas
Как преобразовать Pandas DataFrame в массив NumPy