Python Basics and Matrix Diagonal | Dataford Interview Questions

Describe how you would write a program to find the diagonal of a matrix. Given a non-empty square matrix, implement get_diagonal(matrix) to return a list containing the elements from the top-left to the bottom-right where the row and column indices match. For example, [[1, 2], [3, 4]] returns [1, 4], while [[5]] returns [5].

Constraints

1 <= matrix.length <= 10^4
matrix is square
Each matrix row contains exactly matrix.length elements
-10^9 <= matrix[i][j] <= 10^9

Function Signature

def get_diagonal(matrix):

Your question is Python Basics and Matrix Diagonal. Start with the requirements in the Question tab.

Run and submit as often as you like. When you're ready, talk me through your approach or go straight to the code.

Problem

Constraints

1 <= matrix.length <= 10^4
matrix is square
Each matrix row contains exactly matrix.length elements
-10^9 <= matrix[i][j] <= 10^9

Function Signature

def get_diagonal(matrix):

Your question is Python Basics and Matrix Diagonal. Start with the requirements in the Question tab.

Run and submit as often as you like. When you're ready, talk me through your approach or go straight to the code.