ma.
diag
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.
numpy.diag
See also
Equivalent function for ndarrays.