Python numpy.mask_indices()函数
numpy.mask_indices()函数返回访问(n, n)数组的索引,给定一个屏蔽函数。
语法: numpy.mask_indices(n, mask_func, k = 0)
参数 :
n : [int] 返回的索引将有效地访问形状为(n, n)的数组。
mask_func : [可调用] 一个函数,其调用签名与triu, tril相似。
k : [标量] 一个可选的参数,被传递给mask_func。
返回 : [数组元组] 与mask_func(np.ones((n, n)), k)为True的位置相对应的n个数组的索引。
代码#1:
输出 :
代码#2:
输出 :