np Where in Python - Naukri Code 360