Как добавить столбец в массив 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

Замечательно! Вы успешно подписались.
Добро пожаловать обратно! Вы успешно вошли
Вы успешно подписались на кодкамп.
Срок действия вашей ссылки истек.
Ура! Проверьте свою электронную почту на наличие волшебной ссылки для входа.
Успех! Ваша платежная информация обновлена.
Ваша платежная информация не была обновлена.