Python Numpy Apply Function To Each Element In An