Вы можете использовать один из следующих методов, чтобы добавить столбец в массив 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