[python] NumPy에서 배열을 조작하고 결합하는 방법은 무엇인가요?

배열 연산

NumPy에서는 배열을 기반으로 다양한 연산을 수행할 수 있습니다. 예를 들어, 배열의 모든 요소에 상수를 더하거나, 요소 간의 합, 차, 곱, 나눗셈을 계산할 수 있습니다. 아래는 몇 가지 예시입니다:

상수 더하기

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
result = arr + 10
print(result)  # 출력: [11 12 13 14 15]

요소 간의 합 구하기

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
result = arr1 + arr2
print(result)  # 출력: [5 7 9]

배열 결합

NumPy를 사용하면 배열을 결합하여 더 큰 배열을 생성할 수 있습니다. 배열을 수직 또는 수평으로 결합하는 두 가지 방법이 있습니다.

수직으로 결합하기

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
result = np.vstack((arr1, arr2))
print(result)
"""
출력:
[[1 2 3]
 [4 5 6]]
"""

수평으로 결합하기

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
result = np.hstack((arr1, arr2))
print(result)  # 출력: [1 2 3 4 5 6]

이 외에도 NumPy에서는 배열을 슬라이싱하고 변경하는 등 다양한 배열 조작 기능을 제공합니다. 자세한 내용은 NumPy 공식 문서를 참조하십시오.

NumPy 공식 문서