Às vezes, precisamos encontrar a soma dos elementos diagonais superior direito, superior esquerdo, inferior direito ou inferior esquerdo. Numpy nos fornece a facilidade de calcular a soma de diferentes elementos diagonais usando os métodos numpy.trace() e numpy.diagonal() .

Método 1: Encontrar a soma dos elementos diagonais usando numpy.trace()

Sintaxe: numpy.trace (a, offset = 0, axis1 = 0, axis2 = 1, dtype = None, out = None)  

Exemplo 1: para matriz Numpy 3X3

import numpy as np 
n_array = np.array([[55, 25, 15], 
                    [30, 44, 2], 
                    [11, 45, 77]]) 
print("Numpy Matrix is:") 
print(n_array) 
trace = np.trace(n_array) 
  
  
print("\nTrace of given 3X3 matrix:") 
print(trace) 

Resultado:

Exemplo 2: Para matriz 4X4 Numpy

import numpy as np 
n_array = np.array([[55, 25, 15, 41], 
                    [30, 44, 2, 54], 
                    [11, 45, 77, 11], 
                    [11, 212, 4, 20]]) 
print("Numpy Matrix is:") 
print(n_array) 
trace = np.trace(n_array) 
  
  
print("\nTrace of given 4X4 matrix:") 
print(trace) 

Resultado:

Método 2: Encontrar a soma dos elementos diagonais usando numpy.diagonal()

Sintaxe:

numpy.diagonal (a, deslocamento = 0, eixo 1 = 0, eixo 2 = 1

Exemplo 1: Para Matriz Numpy 3X3

import numpy as np 
n_array = np.array([[55, 25, 15], 
                    [30, 44, 2], 
                    [11, 45, 77]]) 
print("Numpy Matrix is:") 
print(n_array) 
diag = np.diagonal(n_array) 
  
print("\nDiagonal elements are:") 
print(diag) 
  
print("\nSum of Diagonal elements is:") 
print(sum(diag)) 

Resultado:

Exemplo 2: Para Matriz Numpy 5X5

import numpy as np 
n_array = np.array([[5, 2, 1, 4, 6], 
                    [9, 4, 2, 5, 2], 
                    [11, 5, 7, 3, 9], 
                    [5, 6, 6, 7, 2], 
                    [7, 5, 9, 3, 3]]) 
print("Numpy Matrix is:") 
print(n_array) 
diag = np.diagonal(n_array) 
  
print("\nDiagonal elements are:") 
print(diag) 
  
print("\nSum of Diagonal elements is:") 
print(sum(diag)) 

Resultado: