Numpy 배열의 인덱스가 불리언이면
np.array 배열의 인덱스로 불리언 값(True, False)을 사용하면 True에 해당하는 원소만 선택하여 작업할 수 있습니다. 이번 포스팅은 배열의 인덱스가 불리언인 경우를 예를 통해서 알아보겠습니다. 먼저, 아래와 같이 정수로 이뤄진 배열가 있다고 가정해 보겠습니다. x = np.array([1, 2, 3, 4, 5]) 이제, 각 원소가 3보다 큰지를 판단한 불리언 리스트를 만들어 보겠습니다 (즉, x > 3 인지). 결과가 mask = [False, False, False, True, True]이 돼야 하는데, 아래와 같이 판단을 하는 코드를 만들 수 있습니다. import numpy as np mask = [] # 빈 리스트 생성 x = np.array([1,2,3,4,5]) for i in..
2023. 4. 24.