v Calculate The Trace Of A Matrix - Machine Learning

Calculate The Trace Of A Matrix

Preliminaries

# Load library
import numpy as np

Create Matrix

# Create matrix
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

Calculate The Trace

# Calculate the tracre of the matrix
matrix.diagonal().sum()
15