Функция на Python

Нужно написать функцию, условие: дано поле судоку 9 на 9 ( ну то есть, матрица), обращаясь к элементу из поля нужно получить диагональ, в которой он лежит.
Есть доктесты, по которым можно понять как работает функция и проверить, и шаблон есть. Вставляю шаблон сюда.
def get_diagonal(grid: tp.List[tp.List[str]], pos: tp.Tuple[int, int]) -> tp.List:
"""Возвращает все значения диагоналей, в который попадает позиция pos
>>> grid = [["0", "2", "3", "4", "5", "6", "7", "8", "1"], \
["1", "0", "3", "4", "5", "6", "7", "1", "9"], \
["1", "2", "0", "4", "5", "6", "1", "8", "9"], \
["1", "2", "3", "0", "5", "1", "7", "8", "9"], \
["1", "2", "3", "4", "0", "6", "7", "8", "9"], \
["1", "2", "3", "1", "5", "0", "7", "8", "9"], \
["1", "2", "1", "4", "5", "6", "0", "8", "9"], \
["1", "1", "3", "4", "5", "6", "7", "0", "9"], \
["1", "2", "3", "4", "5", "6", "7", "8", "0"]]
>>> get_diagonal(grid, (0, 0))
['0', '0', '0', '0', '0', '0', '0', '0', '0']
>>> get_diagonal(grid, (1, 1))
['0', '0', '0', '0', '0', '0', '0', '0', '0']
>>> get_diagonal(grid, (3, 3))
['0', '0', '0', '0', '0', '0', '0', '0', '0']
>>> get_diagonal(grid, (8, 8))
['0', '0', '0', '0', '0', '0', '0', '0', '0']
>>> get_diagonal(grid, (7, 7))
['0', '0', '0', '0', '0', '0', '0', '0', '0']
>>> get_diagonal(grid, (8, 0))
['1', '1', '1', '1', '0', '1', '1', '1', '1']
>>> get_diagonal(grid, (2, 6))
['1', '1', '1', '1', '0', '1', '1', '1', '1']
>>> get_diagonal(grid, (6, 2))
['1', '1', '1', '1', '0', '1', '1', '1', '1']
"""