numpy.ma.diag

ma.diag(v, k=0)[source]

Извлечение диагонали или построение диагональной матрицы.

Эта функция эквивалентна numpy.diag, учитывая значения с маской. Смотрите numpy.diag для подробностей.

См. также

numpy.diag

Эквивалентная функция для nd-массивов.

Примеры

Создание массива с замаскированными отрицательными значениями:

>>> import numpy as np
>>> x = np.array([[11.2, -3.973, 18], [0.801, -1.41, 12], [7, 33, -12]])
>>> masked_x = np.ma.masked_array(x, mask=x < 0)
>>> masked_x
masked_array(
  data=[[11.2, --, 18.0],
        [0.801, --, 12.0],
        [7.0, 33.0, --]],
  mask=[[False,  True, False],
        [False,  True, False],
        [False, False,  True]],
  fill_value=1e+20)

Изоляция главной диагонали из массива с маской:

>>> np.ma.diag(masked_x)
masked_array(data=[11.2, --, --],
             mask=[False,  True,  True],
       fill_value=1e+20)

Изоляция первой диагонали ниже главной диагонали:

>>> np.ma.diag(masked_x, -1)
masked_array(data=[0.801, 33.0],
             mask=[False, False],
       fill_value=1e+20)