numpy.trace

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

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

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

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

Параметры:

a : array_like Массив входных данных, из которого берутся диагонали. offset : int, optional Смещение диагонали от главной диагонали. Может быть как положительным, так и отрицательным. По умолчанию равно 0. axis1, axis2 : int, optional Оси, которые должны использоваться в качестве первой и второй оси двумерных подмассивов, из которых должны быть взяты диагонали. По умолчанию это две первые оси массива a. dtype : dtype, optional Определяет тип данных возвращаемого массива и аккумулятора, в котором суммируются элементы. Если dtype имеет значение None, а a имеет целочисленный тип с точностью меньше, чем по умолчанию, то используется точность по умолчанию для целочисленных типов. В противном случае точность такая же, как у a. out : ndarray, optional Массив, в который помещается результат. Его тип сохраняется, и он должен иметь правильную форму для хранения результата.

Возвращает:

sum_along_diagonals : ndarray Если a двумерный, возвращается сумма по диагонали. Если 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)