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 :
import numpy as geek
gfg = geek.mask_indices( 3 , geek.triu)
print (gfg)
|
Output :
(array([0, 0, 0, 1, 1, 2]), array([0, 1, 2, 1, 2, 2]))
Code #2 :
import numpy as geek
gfg = geek.mask_indices( 3 , geek.triu, 1 )
print (gfg)
|
Output :
(array([0, 0, 1]), array([1, 2, 2]))
Last Updated :
22 Apr, 2020
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...