https://www.machinelearningplus.com/python/101-numpy-exercises-python/