結果 : square function in python numpy