Python Numpy Apply Function To Each Element