[파이썬][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]]

브로드캐스팅은 크기가 서로 다른 배열 간에도 연산을 자동으로 확장하여 수행할 수 있게 해주므로, 배열 간의 연산을 더 편리하게 수행할 수 있습니다.