本题注意点: 刚从LeetCode500学会all的用法,现学现卖。
https://blog.csdn.net/zuocuomiao/article/details/82177557
import numpy as np
def isToeplitzMatrix(matrix):
"""
:type matrix: List[List[int]]
:rtype: bool
"""
return all(matrix[i][j] == matrix[i+1][j+1] for i in range(len(matrix)-1) for j in range(len(matrix[0])-1))
# samples:
matrix = [
[1,2,3,4],
[5,1,2,3],
[9,5,1,2]
]
print(isToeplitzMatrix(matrix))
matrix = [
[1,2],
[2,2]
]
print(isToeplitzMatrix(matrix))