numpy.where()

welcome to shbcf.ru