Диагональ со сдвигом в NumPy
Функция diag с параметром
k выбирает линию параллельную
главной. При k=1 берут диагональ
на одну ячейку выше главной. Чтение
только главной диагонали уже было
в основной книге; здесь смотрим
смещённые линии.
Из таблицы 3×4 прочитаем
диагональ с k=1 и выведем ряд:
import numpy as np
mat = np.arange(12).reshape(3, 4)
above = np.diag(mat, k=1)
print(above) # выведет [1 6 11]
Из таблицы np.arange(16).reshape(4, 4)
прочитайте диагональ с сдвигом
2 вверх и выведите длину
полученного ряда.
Из таблицы [[0, 1, 2], [3, 4, 5], [6, 7, 8]]
прочитайте диагональ на одну ячейку
ниже главной и выведите первый
элемент ряда.
