numpy.ma.diag — NumPy v1.13 Manual (original) (raw)

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

Extract a diagonal or construct a diagonal array.

This function is the equivalent of numpy.diag that takes masked values into account, see numpy.diag for details.

See also

numpy.diag

Equivalent function for ndarrays.