Unit Array Numpy Python W3schools Lambda Function