Numpy Square An Array