Pergunta de entrevista da empresa Waze

I was asked to solve in python the leetcode question of Toeplitz matrix.