Pythonでサブリストの2番目の要素に基づいてリストをソートする方法

リストが与えられたとき、そのサブリストの2番目の要素に基づいてリスト全体をソートするのが本記事の課題です。ここでは、最も基本的なソートアルゴリズムであるバブルソートを使って実装していきます。

実行例

入力
[['CCC', 15], ['AAA', 10], ['RRRR', 2],['XXXX', 150]]
出力
[['RRRR', 2], ['AAA', 10], ['CCC', 15], ['XXXX', 150]]

アルゴリズムの手順

ステップ1: リストを受け取る。
ステップ2: ネストされたループを使って、サブリストの2番目の要素にアクセスする。
ステップ3: 配列のすべての要素を走査する。
ステップ4: 末尾のi個の要素はすでに正しい位置に配置されている。
ステップ5: 配列を0から n-i-1 の範囲で走査する。
ステップ6: 現在の要素が次の要素より大きい場合は、両者を入れ替える。

サンプルコード

以下のコードでは、3つ目の変数（一時変数）を使用してリスト内の要素を直接入れ替える「インプレース」方式でソートを行っています。

# Python program to sort the lists using the second element of sublist
# In place way to sort, use of third variable.
def sortlist(A):
   l = len(A)
   for i in range(0, l):
      for j in range(0, l-i-1):
         if (A[j][1] > A[j + 1][1]):
            tempo = A[j]
            A[j]= A[j + 1]
            A[j + 1]= tempo
   return A
   # Driver Code
   A =[['AAA', 10], ['CCC', 15], ['RRRR', 2], ['XXXX', 150]]
   print(sortlist(A))

実行結果

[['RRRR', 2], ['AAA', 10], ['CCC', 15], ['XXXX', 150]]

補足：より簡潔な代替手段

実際の開発現場では、Pythonの組み込み関数 sorted() やリストの sort() メソッドに key 引数を指定する方法がより一般的です。例えば、sorted(A, key=lambda x: x[1]) と書くだけで、同じ結果を1行で得ることができます。ただし、ソートアルゴリズムの仕組みを学ぶ目的であれば、上記のようなバブルソートの自前実装は非常に良い練習になります。