Numpy Where Multidimensional Array Example