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

4 votes132 views1 answer
How to print single column in 2d array in python
Python
103 votes210 views6 answers
Time.sleep() python
Python
21 votes192 views6 answers
Last element of list python
Python
22 votes162 views3 answers
Python check if attribute exists in dictionary
Python
0 vote155 views1 answer
Keyboard input raspberry pi
Python
5 votes218 views2 answers
Python windows notification
Python
30 votes178 views3 answers
Iterating index array python
Python
39 votes172 views3 answers
How to write to the end of a file in python
Python
1 vote260 views1 answer
Replace none with empty string python
Python
0 vote164 views1 answer
Azure functions read only file system
Python