numpy.take

numpy.take 함수는 인덱스를 이용해서 어레이의 요소를 가져옵니다.

예제1

import numpy as np

a = [4, 3, 5, 7, 6, 8]
indices = [0, 1, 4, -1]

print(np.take(a, indices))
[4 3 6 8]

어레이로부터 인덱스 [0, 1, 4, -1] 위치에 있는 성분인 [4, 3, 6, 8]의 어레이를 반환합니다.



예제2

import numpy as np

a = np.array([4, 3, 5, 7, 6, 8])
indices = [0, 1, 4, -1]
print(a[indices])
[4 3 6 8]

만약 어레이 a가 NumPy의 ndarray라면, 위의 예제와 같이 a[indices]와 같은 형태로도 인덱스에 해당하는 값들의 어레이를 얻을 수 있습니다.



예제3

import numpy as np

a = [4, 3, 5, 7, 6, 8]
indices = [[0, 1], [4, -1]]

print(np.take(a, indices))
[[4 3]
 [6 8]]

만약 인덱스의 어레이가 1차원이 아니라면, 반환되는 어레이도 같은 차원을 갖습니다.



이전글/다음글

이전글 :
다음글 :