Python 3: Получить диагональные элементы матрицы (списка списков) около точки в этой матрице без NumPy - PullRequest
Новая
       81

Python 3: Получить диагональные элементы матрицы (списка списков) около точки в этой матрице без NumPy

1 голос
/ 03 марта 2020

У меня есть матрица (сформированная из списка списков), которая будет выглядеть примерно так: 

matrix = [[0, 0, 0, 0, 5],
          [0, 0, 0, 4, 0],
          [2, 0, 3, 0, 0],
          [3, 2, 0, 2, 0],
          [1, 0, 2, 0, 1]]

Я пытаюсь создать функцию, которая будет принимать эту матрицу в качестве входных данных вместе с положение в матрице, представленное кортежем, и возвращает две диагонали, которые пересекают эту точку ( без использования NumPy). Например, 

def getDiagonal(matrix, pos)
    (row, col) = pos
    # Smart diagonal finder code #
    return (diag1, diag2)

diagonals = getDiagonals(matrix, (1, 1))
print(diagnonal[0])
print(diagnonal[1])

print(' ')

diagonals = getDiagonals(matrix, (1, 3))
print(diagnonal[0])
print(diagnonal[1])

Ожидаемый результат: 

OUT: [5, 4, 3, 2, 1]
OUT: [2, 2, 2]
OUT: 
OUT: [0, 2, 2]
OUT: [0, 0, 3, 2, 1]

Стоит отметить, что я не против, в каком направлении (снизу вверх или сверху вниз) ) возвращаемые элементы диагоналей. Их можно легко сделать одним способом и почитать, используя reverse(), если это необходимо.

Я рассматривал похожие вопросы, такие как этот , но это в основном касается получения ведущих диагоналей матрица и предоставляет меньше информации о получении диагоналей о точке.

Большое спасибо за вашу помощь и комментарии заранее!

1 Ответ

2 голосов
/ 03 марта 2020

Немного смущает, но я думаю, что это делает: 

def getDiagonals(matrix, pos):
    row, col = pos
    nrows = len(matrix)
    ncols = len(matrix[0]) if nrows > 0 else 0
    # First diagonal
    d1_i, d1_j = nrows - 1 - max(row - col, 0), max(col - row, 0)
    d1_len = min(d1_i + 1, ncols - d1_j)
    diag1 = [matrix[d1_i - k][d1_j + k] for k in range(d1_len)]
    # Second diagonal
    t = min(row, ncols - col - 1)
    d2_i, d2_j = nrows - 1 - row + t, col + t
    d2_len = min(d2_i, d2_j) + 1
    diag2 = [matrix[d2_i - k][d2_j - k] for k in range(d2_len)]
    return (diag1, diag2)

# Test
matrix = [[0, 0, 0, 0, 5],
          [0, 0, 0, 4, 0],
          [2, 0, 3, 0, 0],
          [3, 2, 0, 2, 0],
          [1, 0, 2, 0, 1]]
diagonals = getDiagonals(matrix, (1, 1))
print(diagonals[0])
# [1, 2, 3, 4, 5]
print(diagonals[1])
# [2, 2, 2]

diagonals = getDiagonals(matrix, (1, 3))
print(diagonals[0])
# [2, 2, 0]
print(diagonals[1])
# [1, 2, 3, 0, 0]

diagonals = getDiagonals(matrix, (2, 2))
print(diagonals[0])
# [1, 2, 3, 4, 5]
print(diagonals[1])
# [1, 2, 3, 0, 0]
...