Просте пояснення осей numpy (з прикладами)
Для багатьох функцій у NumPy потрібно вказати вісь, уздовж якої застосовуватиметься певне обчислення.
Як правило, застосовується таке емпіричне правило:
- axis=0 : застосувати обчислення «на стовпець».
- axis=1 : Застосуйте обчислення «на лінію».
На наступному зображенні показано візуальне представлення осей на матриці NumPy із 2 рядками та 4 стовпцями:
У наведених нижче прикладах показано, як використовувати аргумент осі в різних сценаріях із такою матрицею NumPy:
import numpy as np
#create NumPy matrix
my_matrix = np. matrix ([[1, 4, 7, 8], [5, 10, 12, 14]])
#view NumPy matrix
my_matrix
matrix([[ 1, 4, 7, 8],
[5, 10, 12, 14]])
Приклад 1. Знаходження середнього по різних осях
Ми можемо використовувати axis=0 , щоб знайти середнє значення кожного стовпця матриці NumPy:
#find mean of each column in matrix
n.p. mean (my_matrix, axis= 0 )
matrix([[ 3. , 7. , 9.5, 11. ]])
Вихідні дані показують середнє значення кожного стовпця матриці.
Наприклад:
- Середнє значення першого стовпця дорівнює (1 + 5) / 2 = 3 .
- Середнє значення другого стовпця дорівнює (4 + 10) / 2 = 7 .
І так далі.
Ми також можемо використовувати axis=1 , щоб знайти середнє значення кожного рядка в матриці:
#find mean of each row in matrix
n.p. mean (my_matrix, axis= 1 )
matrix([[ 5. ],
[10.25]])
Вихідні дані показують середнє значення кожного рядка матриці.
Наприклад:
- Середнє значення в першому рядку дорівнює (1+4+7+8) / 4 = 5 .
- Середнє значення у другому рядку дорівнює (5+10+12+14) / 4 = 10,25 .
Приклад 2: Знаходження суми по різних осях
Ми можемо використовувати axis=0 , щоб знайти суму кожного стовпця матриці:
#find sum of each column in matrix
n.p. sum (my_matrix, axis= 0 )
matrix([[ 6, 14, 19, 22]])
Вихідні дані показують суму кожного стовпця матриці.
Наприклад:
- Сума першого стовпця дорівнює 1 + 5 = 6 .
- Сума другого стовпця дорівнює 4 + 10 = 14 .
І так далі.
Ми також можемо використовувати axis=1 , щоб знайти суму кожного рядка матриці:
#find sum of each row in matrix
n.p. sum (my_matrix, axis= 1 )
matrix([[20],
[41]])
Вихідні дані показують суму кожного рядка матриці.
Наприклад:
- Сума першого рядка дорівнює 1+4+7+8 = 20 .
- Сума другого рядка дорівнює 5+10+12+14 = 41 .
Додаткові ресурси
У наступних посібниках пояснюється, як виконувати інші типові операції в NumPy:
Як створити матрицю NumPy з випадковими числами
Як нормалізувати матрицю NumPy
Як додати рядок до матриці в NumPy