https://codeforlife.xyz/numpy-basic-functions-i-python/