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：

# Python program explaining
# numpy.mask_indices() function
  
# importing numpy as geek 
import numpy as geek
  
gfg = geek.mask_indices(3, geek.triu)
  
print (gfg)

输出 :

(array([0, 0, 0, 1, 1, 2]), array([0, 1, 2, 1, 2, 2]))

代码#2：

# Python program explaining
# numpy.mask_indices() function
  
# importing numpy as geek 
import numpy as geek
  
gfg = geek.mask_indices(3, geek.triu, 1)
  
print (gfg)

输出 :

(array([0, 0, 1]), array([1, 2, 2]))