Python Numpy Square Root Function