След по выбранным осям в NumPy
Для трёхмерного массива
trace с axis1 и
axis2 суммирует совпадающие
индексы внутри двух осей. В основной
книге след уже был для плоской
таблицы по главной диагонали.
Создадим массив формы (2, 2, 2):
на первом слое сумма 0+2 и 1+3,
на втором 4+6 и 5+7. Просуммируем
диагональ по первым двум осям:
import numpy as np
t = np.arange(8).reshape(2, 2, 2)
print(np.trace(t, axis1=0, axis2=1)) # выведет [6 8]
Для массива формы (2, 2, 2)
из чисел 0…7 по порядку
выведите след по осям 1
и 2.
Для того же массива выведите
след по осям 0 и 2
и сравните длину результата.
