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

0 vote157 views1 answer
Create pdf in python
Python
1 vote157 views1 answer
Pandas groupby size column name
Python
0 vote142 views1 answer
Flask windows auto reload
Python
4 votes138 views1 answer
F-string ponto decimal python
Python
-1 vote301 views1 answer
Combination of 1 2 3 4 5 python
Python
-2 votes143 views1 answer
Turtle opacity
Python
1 vote185 views1 answer
Apolatrix
Python
0 vote153 views1 answer
AttributeError: module 'django.db.models' has no attribute 'ArrayField'
Python
0 vote196 views1 answer
Change creation date filesystem py
Python
20 votes157 views1 answer
ParserError: Error tokenizing data. C error: Expected 1 fields in line 87, saw 2
Python