Как сделать обход квадратной матрицы зигзагом в python?

Question

[Захар Орлов]

Есть матрица NxN, нужно сделать обход таким образом и занести элементы в список.

Answer 1

[Валентин]

Без реккурсий набросал на планшете, возможны ощибки, но алгоритм прост - берем по очереди квадратные блоки матрицы и выбираем у них диагональ

matrix = [
   [11, 12, 13, 14],
   [21, 22, 23, 24],
   [31, 32, 33, 34],
   [41, 42, 43, 44],
]

def walk(matrix):
   zigzag = []
   for index in range(1, len(matrix)):
      slice = [i[:index] for i in matrix[:index]]
      diag = [slice[i][len(slice)-i-1] for i in range(len(slice))]
      if len(diag) % 2:
         diag.reverse()
      zigzag += diag

Comments

[Захар Орлов]

Спасибо! Правда брались только первые три диагонали. Сделал еще один такой же for, только с slice = [i[index:] for i in matrix[index:]] и все работает.

[Валентин]

Ну хорошо, просто интерпретатора под рукой не было, поэтому закосячил)

Answer 2

[Sergey Lerg]

rosettacode.org/wiki/Zig-zag_matrix#Python

Comments

[Захар Орлов]

Этот код генерирует матрицу размером N с элементами, отсортированными зигзагом. Мне же нужно пройтись зигзагом по матрице из случайных чисел, записывая их в список.

[Sergey Lerg]

Захар Орлов: Там есть x,y - вы можете их использовать для вывода элементов вашего массива.
Или отсортировать ваш массив полученным зигзаг массивом.