numpy.ma.diag #
- 嘛。diag ( v , k = 0 ) [来源] #
提取对角线或构造对角线数组。
该函数相当于
numpy.diag
考虑掩码值的函数,请参阅numpy.diag
参考资料。也可以看看
numpy.diag
ndarray 的等效函数。
例子
创建一个带有负值屏蔽的数组:
>>> 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)