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

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

Способ 1. Добавьте столбец в конец таблицы.

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

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

 n.p. 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 в DataFrame Pandas
Как преобразовать DataFrame Pandas в массив NumPy