numpy.digitize

입력한 어레이1의 값이 입력한 어레이2의 어느 값에 해당하는지 인덱스를 반환합니다.

예제

import numpy as np

a = np.array([0.23, 5.43, 2.56, 8.89, 1.29, 4.5])
dig = np.array([0, 1, 2, 4, 8])
ind = np.digitize(a, dig)

print(ind)
print(dig[ind - 1])
[1 4 3 5 2 4]
[0 4 2 8 1 4]

관련 페이지


이전글/다음글

이전글 :
다음글 :