In this NumPy tutorial, you’ll learn how to use the NumPy where function in Python.
Python
x
# import numpy moduleimport numpy as np# create a numpy arrayvar = np.array([11, 22, 33, 44, 55])# print the arrayprint(var)# print the index of the array where var greater than 30print(np.where(var > 30))# print the index of the array where var less than 30print(np.where(var < 30))# print the index of the array where var greater than or equal to 30print(np.where(var >= 30))# print the index of the array where var less than or equal to 30print(np.where(var <= 30))# print the index of the array where var equal to 30print(np.where(var == 30))# print the index of the array where var not equal to 30print(np.where(var != 30))Output:
[11 22 33 44 55] (array([2, 3, 4], dtype=int64),) (array([0, 1], dtype=int64),) (array([2, 3, 4], dtype=int64),) (array([0, 1], dtype=int64),) (array([], dtype=int64),) (array([0, 1, 2, 3, 4], dtype=int64),)
Free resources to learn advanced skills: AiHints and CodeAllow
