numpy.mask_indices() function | Python

numpy.mask_indices() function return the indices to access (n, n) arrays, given a masking function.

Syntax : numpy.mask_indices(n, mask_func, k = 0)
Parameters :
n : [int] The returned indices will be valid to access arrays of shape (n, n).
mask_func : [callable] A function whose call signature is similar to that of triu, tril.
k : [scalar] An optional argument which is passed through to mask_func.
Return : [tuple of arrays] The n arrays of indices corresponding to the locations where mask_func(np.ones((n, n)), k) is True.

Code #1 :

filter_none

edit
close

play_arrow

link
brightness_4
code

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

chevron_right


Output :

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

 
Code #2 :

filter_none

edit
close

play_arrow

link
brightness_4
code

# 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)

chevron_right


Output :

(array([0, 0, 1]), array([1, 2, 2]))
My Personal Notes arrow_drop_up

Check out this Author's contributed articles.

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.