[파이썬][Numpy] 크기가 다른 배열을 자동으로 확장하는 방법 예제
크기가 다른 배열을 자동으로 확장하여 연산을 수행하는 방법을 보다 구체적인 예제로 설명하겠습니다.
예제: 크기가 다른 배열을 자동으로 확장하여 연산하기
import numpy as np
# 크기가 다른 두 배열 생성
arr1 = np.array([[1, 2, 3], [4, 5, 6]]) # 크기: (2, 3)
arr2 = np.array([10, 20]) # 크기: (2,)
# 브로드캐스팅을 사용하여 arr2를 arr1의 크기에 자동으로 확장하여 덧셈 수행
result = arr1 + arr2
print(result)
위의 코드에서 arr1
은 크기가 (2, 3)
이고 arr2
는 크기가 (2,)
인 배열입니다. 이 두 배열을 덧셈 연산할 때 브로드캐스팅이 발생합니다. arr2
가 자동으로 arr1
의 크기에 맞게 확장되어 요소별 덧셈이 수행됩니다.
결과는 다음과 같습니다:
[[11 22 33]
[14 25 36]]
브로드캐스팅은 크기가 서로 다른 배열 간에도 연산을 자동으로 확장하여 수행할 수 있게 해주므로, 배열 간의 연산을 더 편리하게 수행할 수 있습니다.