Bagaimanakah Saya Boleh Isih Tatasusunan NumPy mengikut Lajur Tertentu?

Mengisih Tatasusunan NumPy mengikut Lajur

Mengisih tatasusunan NumPy mengikut lajur khusus selalunya boleh menjadi operasi yang diperlukan untuk manipulasi data. Sebagai contoh, pertimbangkan tatasusunan:

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

Andaikan kita mahu menyusun semula baris a berdasarkan nilai dalam lajur kedua. Untuk mencapai matlamat ini, kita boleh menggunakan fungsi argsort() dalam NumPy.

a[:, 1].argsort()

Ungkapan di atas menggunakan argsort() pada lajur kedua a, menghasilkan tatasusunan indeks yang sepadan dengan nilai yang diisih. Secara khusus, ia mengembalikan:

array([1, 0, 2])

Indeks ini mewakili tertib baris a harus disusun semula untuk mengisih mengikut lajur kedua dalam tertib menaik. Menggabungkan ini dengan pengindeksan tatasusunan, kita boleh mendapatkan tatasusunan yang diisih:

a[a[:, 1].argsort()]

Ungkapan ini mengembalikan:

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

Seperti yang dikehendaki, baris a telah diisih mengikut lajur kedua .

Atas ialah kandungan terperinci Bagaimanakah Saya Boleh Isih Tatasusunan NumPy mengikut Lajur Tertentu?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!