Matrix diagonal sum Python

# Problem Link : https://leetcode.com/problems/matrix-diagonal-sum/

class Solution(object):
    def diagonalSum(self, array):
        """
        :type array: List[List[int]]
        :rtype: int
        """
        n = len(array)
        primary = 0
        secondary = 0;
        for i in range(0, n):
            primary += array[i][i]
            secondary += array[i][n-i-1]
        if (n % 2 == 0): return primary + secondary
        else: return primary + secondary - array[n//2][n//2]

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

5 votes195 views2 answers
Exemple python gradient
Python
0 vote204 views1 answer
How to get checkbutton from a list
Python
2 votes201 views1 answer
Unsigned int python
Python
5 votes187 views3 answers
Click a button using selenium python
Python
0 vote214 views1 answer
Dataframe select row by index value
Python
33 votes173 views4 answers
Sort and reverse list in python
Python
42 votes199 views5 answers
Iterate array python with index
Python
3 votes211 views2 answers
Postgres Database Django
Python
18 votes244 views9 answers
Dictionary in python
Python
8 votes222 views2 answers
Combine two dataframe in pandas
Python