función numpy.mask_indices() | Python

numpy.mask_indices()función devuelve los índices para acceder a (n, n) arrays, dada una función de enmascaramiento.

Sintaxis: numpy.mask_indices(n, mask_func, k = 0)
Parámetros:
n: [int] Los índices devueltos serán válidos para acceder a arrays de forma (n, n).
mask_func: [llamable] Una función cuya firma de llamada es similar a la de triu, tril.
k : [escalar] Un argumento opcional que se pasa a mask_func.
Devuelve: [tupla de arrays] Las n arrays de índices correspondientes a las ubicaciones donde mask_func(np.ones((n, n)), k) es True.

Código #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)

Producción :

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

 
Código #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)

Producción :

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