Python Numpy.square()-정사각형

  1. numpy.square() 구문
  2. 예제 코드: numpy.square()
  3. 예제 코드: out 파라미터가 있는 numpy.square()
  4. 예제 코드: 음수가있는numpy.square()
  5. 예제 코드: 복소수가있는numpy.square()

Numpy.square() 함수는 주어진 배열에있는 모든 요소의 제곱을 계산합니다.

Numpy.sqrt() 메서드의 역 연산입니다.

numpy.square() 구문

numpy.square(arr,
          out=None) 

매개 변수

arr 입력 배열
out out이 주어지면 결과는out에 저장됩니다. outarr와 같은 모양이어야합니다.

반환

out이 주어 지더라도 입력 배열에있는 각 요소의 제곱 배열을 반환합니다.

예제 코드: numpy.square()

import numpy as np

arr = [1, 3, 5, 7]

arr_sq = np.square(arr)

print(arr_sq)

산출:

[ 1  9 25 49]

예제 코드: out 파라미터가 있는 numpy.square()

import numpy as np

arr = [1, 3, 5, 7]
out_arr = np.zeros(4)

arr_sq = np.square(arr, out_arr)

print(out_arr)
print(arr_sq)

산출:

[ 1  9 25 49]
[ 1  9 25 49]

out_arrarr와 같은 모양이며arr의 사각형이 저장됩니다.

그리고numpy.square()메서드는 위와 같이 정사각형 배열도 반환합니다.

outarr와 같은 모양이 아니면ValueError를 발생시킵니다.

import numpy as np

arr = [1, 3, 5, 7]
out_arr = np.zeros(3)

arr_sq = np.square(arr, out_arr)

print(out_arr)
print(arr_sq)

산출:

Traceback (most recent call last):
  File "C:\Test\test.py", line 6, in <module>
    arr_sq = np.square(arr, out_arr)
ValueError: operands could not be broadcast together with shapes (4,) (3,) 

예제 코드: 음수가있는numpy.square()

import numpy as np

arr = [-1, -3, -5, -7]

arr_sq = np.square(arr)

print(arr_sq)

산출:

[ 1  9 25 49]

예제 코드: 복소수가있는numpy.square()

import numpy as np

arr = [1+2j, -2-1j, 2-3j, -3+4j]

arr_sq = np.square(arr)

print(arr_sq)

산출:

[-3. +4.j  3. +4.j -5.-12.j -7.-24.j]