data-science-summary/preprocessing

배열 인덱스 정렬 in python

고수트 2020. 10. 30. 22:53
반응형

배열 인덱스 오름차순 정렬하는법

import numpy as np
x = np.array([1,4,3,2])
print(np.argsort(x))

 

배열 인덱스 내림차순 정렬하는법 

import numpy as np
x = np.array([1,4,3,2])
print(np.argsort(x)[::-1])

 

2차원 배열 인덱스 오름차순 정렬하는법

import numpy as np
x = np.array([[1,4,3,2], [5,2,3,1]])
print(np.argsort(x))

 

2차원 배열 인덱스 내림차순 정렬하는법

import numpy as np
x = np.array([[1,4,3,2], [5,2,3,1]])
arr = []
for i in x:
    arr.append(np.argsort(i)[::-1])
np.array(arr)
반응형