使用第二列對二維數組排序，並且如果第二列中的元素按第一列排序

我已經編寫了一個python程序來使用第二列對二維數組進行排序，並且如果第二列中的元素相同按第一列排序。儘管我用我最基本的Python知識解決了這個問題。我認爲它可以改善。任何人都可以幫助優化它？請也建議如果使用其他數據類型進行排序將是不錯的選擇？使用第二列對二維數組排序，並且如果第二列中的元素按第一列排序

#created a two dimensional array 
two_dim_array=[[2, 5], [9, 1], [4, 8], [10, 0], [50, 32], [33, 31],[1, 5], [12, 5], [22, 5], [32, 5], [9, 5],[3, 31], [91, 32] ] 
#get the length of the array 
n_ship=len(two_dim_array) 
#sorting two dimensional array by using second column 
sort_by_second_column=sorted(two_dim_array, key=lambda x: x[1], reverse=False) 
#declared new variable for storing second soeted array 
new_final_data=[] 
#parameter used to slice the two dimensional column 
first_slice=0 
    #tmp=[] 
index=[0] 
for m in range(1, n_ship): 
     #print('m is: '+str(m)+'final_data[m-1][1] is: '+str(final_data[m-1][1])+'final_data[m][1] is: '+str(final_data[m][1])) 
    #subtracting second column elements to detect changes and saved to array 
    if(abs(sort_by_second_column[m-1][1]-sort_by_second_column[m][1])!=0): 
     index.append(m) 
     # print(index) 
     l=1 
# used the above generated index to slice the data 
for z in range(len(index)): 
    tmp=[] 
    if(l==1): 
     first_slice=0 
     last=index[z+1] 
     mid_start=index[z] 
     # print('l is start'+ 'first is '+str(first_slice)+'last is'+str(last)) 
     v=sort_by_second_column[:last] 

    elif l==len(index): 
     first_slice=index[z] 
     # print('l is last'+str(1)+ 'first is '+str(first_slice)+'last is'+str(last)) 
     v=sort_by_second_column[first_slice:] 
    else: 
     first_slice=index[z] 
     last=index[z+1] 
     #print('l is middle'+str(1)+ 'first is '+str(first_slice)+'last is'+str(last)) 
     v=sort_by_second_column[first_slice:last] 


    tmp.extend(v) 
    tmp=sorted(tmp, key=lambda x: x[0], reverse=False) 
     #print(tmp) 
    new_final_data.extend(tmp) 
     # print(new_final_data) 
    l+=1 


for l in range(n_ship): 
     print(str(new_final_data[l][0])+' '+str(new_final_data[l][1])) 


''' Input 
    2 5 
    9 1 
    4 8 
    10 0 
    50 32 
    33 31 
    1 5 
    12 5 
    22 5 
    32 5 
    9 5 
    3 31 
    91 32 

    Output 
    10 0 
    9 1 
    1 5 
    2 5 
    9 5 
    12 5 
    22 5 
    32 5 
    4 8 
    3 31 
    33 31 
    50 32 
    91 32'''

來源

2017-02-01 curious

你應該閱讀文檔上sorted()，因爲這是你需要使用什麼： https://docs.python.org/3/library/functions.html#sorted

newarray=sorted(two_dim_array, key=lambda x:(x[1],x[0]))

輸出：

[10, 0] 
[9, 1] 
[1, 5] 
[2, 5] 
[9, 5] 
[12, 5] 
[22, 5] 
[32, 5] 
[4, 8] 
[3, 31] 
[33, 31] 
[50, 32] 
[91, 32]

來源

2017-02-01 07:39:18 Trelzevir

@Thank你Trelzevir。我會根據你的建議。 – curious

您也可以使用'itemgetter（1，0）'作爲關鍵字，這與lambda實際上是一樣的。 –

使用第二列對二維數組排序，並且如果第二列中的元素按第一列排序

回答

相關問題