NumPy'de dizileri sütuna göre sıralama

NumPy'de bir diziyi n'inci sütuna göre nasıl sıralayabilirim?

Örneğin,

a = array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])

Satırları ikinci sütuna göre sıralamak istiyorum, öyle ki geri alıyorum:

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])
Paul Wintz
Edited question 2018ööp5öö7 Nisan 2018 в 7:05
Programlama
sorting arrays python numpy scipy
2010ösp32ös15 Mayıs 2010 в 3:32
64 görünümler

Sanırım bu işe yarar: a[a[:,1].argsort()]

Bu, a nın ikinci sütununu gösterir ve buna göre sıralar.

makis
Edited answer 2018ööp17öö10 Aralık 2018 в 10:17
Yorumlar (9)
Çözüm

[@steve][1]'s aslında bunu yapmanın en zarif yoludur.

Doğru" yol için [numpy.ndarray.sort][2]'nin sıra anahtar sözcüğü argümanına bakın

Ancak, dizinizi alanları olan bir dizi (yapılandırılmış bir dizi) olarak görmeniz gerekir.

Eğer dizinizi başlangıçta alanlarla tanımlamadıysanız, "doğru" yol oldukça çirkindir...

Hızlı bir örnek olarak, sıralamak ve bir kopyasını döndürmek için:

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]])

Yerinde sıralamak için:


In [6]: a.view('i8,i8,i8').sort(order=['f1'], axis=0) #
Community
Edited answer 2017ösp10ös12 Mayıs 2017 в 12:10
Yorumlar (11)

Python dokümantasyon vikisinden]1, yapabileceğinizi düşünüyorum:

a = ([[1, 2, 3], [4, 5, 6], [0, 0, 1]]); 
a = sorted(a, key=lambda a_entry: a_entry[1]) 
print a

Çıktı şudur:

[[[0, 0, 1], [1, 2, 3], [4, 5, 6]]]
Peter Mortensen
Edited answer 2017ööp0öö10 Mayıs 2017 в 10:00
Yorumlar (1)