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` 함수를 활용해보세요.

반응형