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

2 votes113 views1 answer
How to convert gb to mb in python
Python
1 vote100 views1 answer
Pyspark Aggregation on multiple columns
Python
1 vote112 views1 answer
How to pick out separate columns from the pandas dataframe object
Python
3 votes138 views1 answer
Autoslugfield django 3
Python Django
1 vote121 views1 answer
How to move a column to the beginning in dataframe
Python
10 votes121 views2 answers
Django delete object
Python
4 votes102 views2 answers
Tty escape
Python
2 votes103 views1 answer
Pandas pass two columns to function
Python
0 vote109 views1 answer
Forbidden (CSRF token missing or incorrect.): /extension/stripe-pay/ WARNING 2021-06-01 13:45:22,532 log 408 140165573588736 Forbidden (CSRF token missing or incorrect.): /extension/stripe-pay/
Python
1 vote92 views1 answer
Install quick-mailer
Python