Want to get the index value that meets the condition , Need to use np.where.
1.np.where(condition, x, y)
x： The output of the function when the condition is satisfied
y： Output when the condition is not met
2.np.where(condition)
No, x and y Parameters , Then the list index satisfying the condition is output in the form of tuple .
3.np.argwhere(condition)
Find the indices of array elements that are non-zero, grouped by element.
The returned result is not a tuple , Directly return the coordinate index .
