Try a W3Schools Numpy Exercise here