Calcule la suma de los elementos diagonales de una array NumPy

A veces necesitamos encontrar la suma de los elementos diagonales Superior derecha, Superior izquierda, Inferior derecha o Inferior izquierda. Numpy nos brinda la posibilidad de calcular la suma de diferentes elementos diagonales utilizando los métodos numpy.trace() y numpy.diagonal() .

Método 1: Encontrar la suma de elementos diagonales usando numpy.trace()

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

Ejemplo 1: Para array Numpy 3X3

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

Producción:

Ejemplo 2: Para array Numpy 4X4

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

Producción:

Método 2: Encontrar la suma de elementos diagonales usando numpy.diagonal()

Sintaxis:

numpy.diagonal(a, offset=0, axis1=0, axis2=1

Ejemplo 1: Para 3X3 Numpy Matrix

# importing Numpy package
import numpy as np
  
# creating a 3X3 Numpy matrix
n_array = np.array([[55, 25, 15],
                    [30, 44, 2],
                    [11, 45, 77]])
  
# Displaying the Matrix
print("Numpy Matrix is:")
print(n_array)
  
# Finding the diagonal elements of a matrix
diag = np.diagonal(n_array)
  
print("\nDiagonal elements are:")
print(diag)
  
print("\nSum of Diagonal elements is:")
print(sum(diag))

Producción:

Ejemplo 2: Para 5X5 Numpy Matrix

# importing Numpy package
import numpy as np
  
# creating a 5X5 Numpy matrix
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]])
  
# Displaying the Matrix
print("Numpy Matrix is:")
print(n_array)
  
# Finding the diagonal elements of a matrix
diag = np.diagonal(n_array)
  
print("\nDiagonal elements are:")
print(diag)
  
print("\nSum of Diagonal elements is:")
print(sum(diag))

Producción: