Numpy Python Code Example