Python/numpy
파이썬 numpy.setdiff1d 함수 활용하기
PySun
2024. 8. 24. 06:49
반응형
Numpy setdiff1d 함수 소개:
Numpy의 `setdiff1d` 함수는 첫 번째 배열에서 두 번째 배열에 있는 요소를 제거한 후 남은 요소를 반환하는 함수입니다. 이 함수는 배열 간의 차집합을 계산할 때 유용하게 사용됩니다.
기본 사용법:
`setdiff1d` 함수를 사용하여 두 배열의 차집합을 계산하는 방법은 다음과 같습니다.
import numpy as np
# 두 배열 생성
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([3, 4, 5, 6, 7])
# 차집합 계산
result = np.setdiff1d(arr1, arr2)
print("첫 번째 배열에서 두 번째 배열의 요소를 제거한 차집합:", result)
위의 예시 코드에서는 `np.array([1, 2, 3, 4, 5])`와 `np.array([3, 4, 5, 6, 7])`를 사용하여 두 개의 배열을 생성하고, `setdiff1d` 함수를 사용하여 첫 번째 배열에서 두 번째 배열의 요소를 제거한 차집합을 계산합니다.
결과:
두 배열의 차집합을 찾는데 사용되는 `setdiff1d` 함수를 통해 배열 간의 차이를 쉽게 계산할 수 있습니다. 위의 예시 코드를 참고하여 `setdiff1d` 함수를 활용해보세요.
반응형