結果 : square in python numpy