Numpy - searchsorted

https://numpy.org/doc/stable/reference/generated/numpy.searchsorted.html#numpy.searchsorted

numpy.searchsorted — NumPy v1.19 Manual

 

numpy.searchsorted(a, v, side='left', sorter=None)

a = 1차원 배열, 오름차순으로 정렬되어 있어야함

 

v = 내가 찾고자 하는 값

 

side='left', sorter=None은 기본값

 

 

 

 

 

[예시]

 

 

1, 2, 3, 4, 5에서 3의 위치는 2

 

 

 

 

 

 

 

적절한 인덱스를 찾지 못하면 0 또는 배열의 길이를 리턴한다.

 

-10 → 0

10 → 5

 

2와 3의 인덱스는 정상적으로 리턴됨

 

 

 

 

 

 

random을 이용한 예시

 

 

0, 1사이의 분포로 랜덤한 ndarray 생성

 

 

 

누적합을 이용해 자동 정렬

 

 

 

 

이제 예를 들어 처음으로 4가 나오는 위치를 얻고 싶다면

searchsorted 메서드를 이용한다.

 

4.11083803의 위치인 6을 리턴

 

단, 이는 0부터 시작하는 인덱스이므로

 

 

 

이처럼 1을 더해서 사용해야할 경우도 있을 수 있다.