numpy.trace

numpy.trace(a, offset=0, axis1=0, axis2=1, dtype=None, out=None)[source]

Возвращает сумму по диагоналям массива.

Если a является 2-мерным, возвращается сумма по диагонали с заданным смещением, то есть сумма элементов a[i,i+offset] для всех i.

Если a имеет более двух измерений, то для определения 2-мерных подмассивов, чьи следы нужно вернуть, используются оси axis1 и axis2. Форма результирующего массива совпадает с формой a с удаленными axis1 и axis2.

Parameters

aarray_like

Входной массив, из которого берутся диагонали.

offsetint, optional

Смещение диагонали от главной диагонали. Может быть положительным и отрицательным. По умолчанию 0.

axis1, axis2int, optional

Оси, которые используются в качестве первой и второй оси 2-мерных подмассивов, из которых должны быть взяты диагонали. По умолчанию используются первые две оси a.

dtypedtype, optional

Определяет тип данных возвращаемого массива и аккумулятора, где суммируются элементы. Если dtype имеет значение None и a имеет целочисленный тип с разрядностью меньше, чем по умолчанию, то используется по умолчанию целочисленная разрядность. В противном случае разрядность такая же, как у a.

outndarray, optional

Массив, в который помещается результат. Его тип сохраняется, и он должен иметь правильную форму для хранения результата.

Returns

sum_along_diagonalsndarray

Если a является 2-мерным, возвращается сумма по диагонали. Если a имеет больше измерений, то возвращается массив сумм по диагоналям.

См. также

diag, diagonal, diagflat

Примеры

>>> np.trace(np.eye(3))
3.0
>>> a = np.arange(8).reshape((2,2,2))
>>> np.trace(a)
array([6, 8])

>>> a = np.arange(24).reshape((2,2,2,3))
>>> np.trace(a).shape
(2, 3)