NumPy에서 n 번째 열을 기준으로 배열을 정렬하려면 어떻게합니까?
예를 들어
a = array([[9, 2, 3],
[4, 5, 6],
[7, 0, 5]])두 번째 열을 기준으로 행을 정렬하고 싶습니다.
array([[7, 0, 5],
[9, 2, 3],
[4, 5, 6]])답변
@ steve 의 대답 은 실제로 가장 우아한 방법입니다.
“올바른”방법은 numpy.ndarray.sort 의 order 키워드 인수를 참조하십시오.
그러나 배열을 필드가있는 배열 (구조적 배열)로보아야합니다.
처음에 필드로 배열을 정의하지 않은 경우 “올바른”방법은 매우 추악합니다 …
간단한 예를 들어 정렬하고 사본을 반환하려면 다음을 수행하십시오.
In [1]: import numpy as np
In [2]: a = np.array([[1,2,3],[4,5,6],[0,0,1]])
In [3]: np.sort(a.view('i8,i8,i8'), order=['f1'], axis=0).view(np.int)
Out[3]:
array([[0, 0, 1],
[1, 2, 3],
[4, 5, 6]])제자리에 정렬하려면 다음을 수행하십시오.
In [6]: a.view('i8,i8,i8').sort(order=['f1'], axis=0) #<-- returns None
In [7]: a
Out[7]:
array([[0, 0, 1],
[1, 2, 3],
[4, 5, 6]])@ Steve ‘s는 내가 아는 한 실제로 그것을하는 가장 우아한 방법입니다 …
이 방법의 유일한 장점은 “order”인수가 검색을 정렬 할 필드 목록이라는 것입니다. 예를 들어 order = [ ‘f1’, ‘f2’, ‘f0’]을 제공하여 두 번째 열, 세 번째 열, 첫 번째 열을 기준으로 정렬 할 수 있습니다.
답변
나는 이것이 효과가 있다고 생각한다. a[a[:,1].argsort()]
이것은 두 번째 열을 나타내며 a그에 따라 정렬합니다.
답변
Steve Tjoa의 방법에 따라 mergesort와 같은 안정적인 정렬을 사용하고 인덱스를 가장 중요하지 않은 열에서 가장 중요한 열로 정렬하여 여러 열을 정렬 할 수 있습니다.
a = a[a[:,2].argsort()] # First sort doesn't need to be stable.
a = a[a[:,1].argsort(kind='mergesort')]
a = a[a[:,0].argsort(kind='mergesort')]이것은 열 0, 1, 2로 정렬됩니다.
답변
누군가 프로그램의 중요한 부분에서 정렬을 사용하려는 경우 다른 제안에 대한 성능 비교가 있습니다.
import numpy as np
table = np.random.rand(5000, 10)
%timeit table.view('f8,f8,f8,f8,f8,f8,f8,f8,f8,f8').sort(order=['f9'], axis=0)
1000 loops, best of 3: 1.88 ms per loop
%timeit table[table[:,9].argsort()]
10000 loops, best of 3: 180 µs per loop
import pandas as pd
df = pd.DataFrame(table)
%timeit df.sort_values(9, ascending=True)
1000 loops, best of 3: 400 µs per loop그래서 지금까지 가장 빠른 방법은 argsort 를 사용한 색인 생성입니다 …
답변
에서 파이썬 문서 위키 , 나는 당신이 할 수 있다고 생각 :
a = ([[1, 2, 3], [4, 5, 6], [0, 0, 1]]);
a = sorted(a, key=lambda a_entry: a_entry[1])
print a출력은 다음과 같습니다.
[[[0, 0, 1], [1, 2, 3], [4, 5, 6]]]답변
에서 NumPy와 메일 링리스트 , 여기에 또 다른 솔루션입니다 :
>>> a
array([[1, 2],
[0, 0],
[1, 0],
[0, 2],
[2, 1],
[1, 0],
[1, 0],
[0, 0],
[1, 0],
[2, 2]])
>>> a[np.lexsort(np.fliplr(a).T)]
array([[0, 0],
[0, 0],
[0, 2],
[1, 0],
[1, 0],
[1, 0],
[1, 0],
[1, 2],
[2, 1],
[2, 2]])답변
나는 비슷한 문제가 있었다.
내 문제:
SVD를 계산하고 고유 값 을 정렬해야합니다. 을 내림차순 . 그러나 고유 값과 고유 벡터 간의 매핑을 유지하고 싶습니다. 내 고유 값은 첫 번째 행에 있었고 해당 고유 벡터는 같은 열에있었습니다.
그래서 첫 번째 행을 기준으로 2 차원 배열을 내림차순으로 정렬합니다.
내 솔루션
a = a[::, a[0,].argsort()[::-1]]어떻게 작동합니까?
a[0,] 정렬하려는 첫 번째 행입니다.
이제 argsort를 사용하여 인덱스 순서를 가져옵니다.
[::-1]내림차순이 필요 하기 때문에 사용 합니다.
마지막으로 a[::, ...]올바른 순서로 열이있는보기를 얻는 데 사용 합니다.
