Answers for "how to find diagonal elements of a matrix in python"

1

python create a matrix with one in diagonal

# Create a matrix in python and fill
import numpy as np
a = np.zeros((3, 3), int) # Create matrix with only 0
np.fill_diagonal(a, 1) # fill diagonal with 1
print(a)
Posted by: Guest on January-02-2021
0

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]
Posted by: Guest on January-09-2022

Code answers related to "how to find diagonal elements of a matrix in python"

Python Answers by Framework

Browse Popular Code Answers by Language