w3resource

NumPy: Compute the sum of the diagonal element of a given array

NumPy: Linear Algebra Exercise-15 with Solution

Write a NumPy program to compute the sum of the diagonal element of a given array.

Sample Solution:

Python Code :

import numpy as np
m = np.arange(6).reshape(2,3)
print("Original matrix:")
print(m)
result =  np.trace(m)
print("Condition number of the said matrix:")
print(result)

Sample Output:

Original matrix:
[[0 1 2]
 [3 4 5]]
Condition number of the said matrix:
4

Python Code Editor:


Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a NumPy program to compute the condition number of a given matrix.
Next: Write a NumPy program to get the lower-triangular L in the Cholesky decomposition of a given array.

What is the difficulty level of this exercise?

Test your Python skills with w3resource's quiz



Python: Tips of the Day

Python: The Zip() Function

>>> students = ('John', 'Mary', 'Mike')
>>> ages = (15, 17, 16)
>>> scores = (90, 88, 82, 17, 14)
>>> for student, age, score in zip(students, ages, scores):
...     print(f'{student}, age: {age}, score: {score}')
... 
John, age: 15, score: 90
Mary, age: 17, score: 88
Mike, age: 16, score: 82
>>> zipped = zip(students, ages, scores)
>>> a, b, c = zip(*zipped)
>>> print(b)
(15, 17, 16)