如何按行排序數組？

Question

我有一個數組形式的A如下：

A = [[ 4 3 2 1] 
    [ 8 7 6 5] 
    [12 11 10 9] 
    [16 15 14 13]] 

我想排序這個數組按行。所需的輸出是：

A = [[ 1 2 3 4] 
    [ 5 6 7 8] 
    [ 9 10 11 12] 
    [13 14 15 16]] 

我使用下面的代碼（但它不工作）嘗試：

import numpy as np 
A = np.array([[4, 3, 2, 1], [8, 7, 6, 5], [12, 11, 10, 9], [16, 15, 14, 13]]) 
print A[np.lexsort(A, axis = 1)] 

我怎麼排序這個數組按行？

Answer 1

目前尚不清楚你的意思是「按行排序」，但你似乎想在陣列，您可以通過遍歷它和排序每個元素做的元素進行排序：

for i in A: i.sort() 

如果你也想外陣列本身進行排序，然後，嗯，有點吧：

A.sort() 

Answer 2

A = [[ 4, 3, 2, 1], 
    [12, 11, 10, 9], 
    [ 8, 7, 6, 5], 
    [16, 15, 14, 13]] 
for i in A: 
    i.sort() 

A.sort() 
print A 

我認爲逐行意味着你也想排序外列表

Answer 3

兩個不同的答案，無論是單行，如下：

sorted_A_1 = sorted(sorted(i) for i in A) 
sorted_A_2 = [sorted(i) for i in sorted(A)] 

雖然你可能想簡單地修改現有的名單，這些都產生一個新的列表。第一個管道將A中的每個列表排序到排序函數的結果，第二個只是列表理解。另外，雖然這些可能並不比已經發布的答案更清楚，但我個人認爲，看到解決給定問題的不同方法是有幫助的！

Answer 4

Numpy的sort有一個axis參數，可用於指定哪個軸被排序。

import numpy as np 
A = np.array([[4, 3, 2, 1], [8, 7, 6, 5], [12, 11, 10, 9], [16, 15, 14, 13]]) 

A.sort(axis=1) 

這給：

[[ 1 2 3 4] 
[ 5 6 7 8] 
[ 9 10 11 12] 
[13 14 15 16]] 

sort默認使用axis作爲數組的最後一個軸，所以這裏會1，所以你並不需要在這種情況下，指定axis=1 。