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 vote169 views1 answer
Cairo.context transform vertical text python
Python
0 vote199 views1 answer
How to access rows and columns indexing numpy
Python
13 votes191 views5 answers
Group by pandas count
Python
0 vote244 views1 answer
Torch.unsqueze
Python
0 vote193 views1 answer
How to find empty rows of a dataset in python
Python
0 vote175 views1 answer
Python to apk converter software download for pc
Python
6 votes173 views2 answers
Plt.imshow grayscale
Python
0 vote181 views1 answer
Young C so new(pro.cashmoneyap x nazz music) soundcloud
Python
0 vote170 views1 answer
Python split string regular expression
Python
0 vote186 views1 answer
Left join two dataframes pandas on two different column names
Python