Python diagonal sum

# A simple Python3 program to find
# sum of diagonals
MAX = 100
 
def printDiagonalSums(mat, n):
 
    principal = 0
    secondary = 0
    for i in range(0, n):
        principal += mat[i][i]
        secondary += mat[i][n - i - 1]
         
    print("Principal Diagonal:", principal)
    print("Secondary Diagonal:", secondary)
 
# Driver code
a = [[ 1, 2, 3, 4 ],
     [ 5, 6, 7, 8 ],
     [ 1, 2, 3, 4 ],
     [ 5, 6, 7, 8 ]]
printDiagonalSums(a, 4)
 
# This code is contributed
# by ihritik

Source: geeksforgeeks.org

Sum of diagonal numpy

arr= np.arange(16)
arr= np.reshape(arr,(4,4))

a=np.diagonal(arr, 0,0,1)

result=sum(a)
print(result)

-1

Source: numpy.org

Similar codes

1 vote113 views1 answer
How to construct simple timedelta in python
Python
4 votes115 views1 answer
Change type of array python
Python
0 vote118 views1 answer
Default dictionary value
Python
5 votes116 views2 answers
Making variable if it is none python
Python
0 vote134 views1 answer
Credential not provided when i try to sign up a new user django
Python Django
2 votes111 views2 answers
How to read csv file online into pandas
Python
0 vote149 views1 answer
Write a Python program to accept two strings as input and check if they are identical copy of each other or if the second string is a substring of the first string.
Python
0 vote119 views1 answer
Printing float number python
Python
0 vote68 views1 answer
Df get columnvalues
Python
16 votes112 views5 answers
Pandas groupby aggregate multiple columns
Python