#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ ENGG1811 Illustrating the numpy where function """ import numpy as np # Problem parameters array1 = np.array([ [-3, 0, 0, 5], [ 6, -4, 6, 7], [ 3, 5, 2, 3]]) # where coordinates_equal_to_7 = np.where(array1 == 7) coordinates_geq_5 = np.where(array1 >= 5)