Using If In Lambda Function In Python Examples Using Numpy