Как добавить строки и столбцы массива numpy
Вы можете использовать следующие методы для добавления строк и столбцов 2D-массива NumPy:
Способ 1: добавить строки массива NumPy
arr. sum (axis= 1 )
Способ 2: добавление столбцов массива NumPy
arr. sum (axis= 0 )
В следующих примерах показано, как использовать каждый метод на практике со следующим двумерным массивом NumPy:
import numpy as np #create NumPy array arr = np. arange ( 18 ). reshape ( 6,3 ) #view NumPy array print (arr) [[ 0 1 2] [3 4 5] [6 7 8] [9 10 11] [12 13 14] [15 16 17]]
Пример 1. Добавьте строки массива NumPy.
Мы можем использовать следующий синтаксис для добавления строк массива NumPy:
import numpy as np #calculate sum of rows in NumPy array arr. sum (axis= 1 ) array([ 3, 12, 21, 30, 39, 48])
В полученной таблице показана сумма каждой строки двумерной таблицы NumPy.
Например:
- Сумма значений в первой строке равна 0 + 1 + 2 = 3 .
- Сумма значений в первой строке равна 3 + 4 + 5 = 12 .
- Сумма значений в первой строке равна 6 + 7 + 8 = 21 .
И так далее.
Пример 2: сумма столбцов таблицы NumPy
Мы можем использовать следующий синтаксис для добавления столбцов массива NumPy:
import numpy as np #calculate sum of columns in NumPy array arr. sum (axis= 0 ) array([45, 51, 57])
В результирующей таблице показана сумма каждого столбца 2D-таблицы NumPy.
Например:
- Сумма значений в первом столбце равна 0+3+6+9+12+15 = 45 .
- Сумма значений в первой строке равна 1+4+7+10+13+16 = 51 .
- Сумма значений в первой строке равна 2+5+8+11+14+17 = 57 .
Примечание . Полную документацию по функции NumPy sum() можно найти здесь .
Дополнительные ресурсы
В следующих руководствах объясняется, как выполнять другие распространенные операции в NumPy:
Как найти индекс значения в массиве NumPy
Как получить определенный столбец из массива NumPy
Как добавить столбец в массив NumPy