numpy.ma.diag#
- ma.diag(v, k=0)[source]#
提取对角线或构造对角线数组。
此函数等效于
numpy.diag
,它考虑了掩码值,有关详细信息,请参阅numpy.diag
。另请参阅
numpy.diag
ndarray 的等效函数。
示例
>>> import numpy as np
创建一个带有负值掩码的数组
>>> 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)