数値を含む文字列のリストを数値順でソートするときに使うPythonのラムダ式

はじめに

Pythonでのプログラミング中に、「数値を含む文字列のリストを数値順でソートする処理」が必要になり、試行錯誤していました。本来の処理に直接関係ない部分なので、可読性を犠牲にしてでもコンパクトにしたくなり、勉強がてらにラムダ式を使ってみました。シンプルに書けたのでメモとして残します。

ラムダ式

変数に代入して関数として使う方法を使いました。

• 変数名 = lambda 引数: 処理

num_in_str = lambda str: [int(c) if c.isdigit() else c for c in re.split(r'(\d+)', str) ]

動作確認

コード

import re

num_in_str = lambda str: [int(c) if c.isdigit() else c for c in re.split(r'(\d+)', str) ]

lists = ['S23', 'S14', 'S7', 'S11', 'S1']
lists.sort(key=num_in_str)
print(lists)

結果

['S1', 'S7', 'S11', 'S14', 'S23']

参考にした情報

数値を含む文字列のリストを数値順でソート (numerical sort, python)

参考にしたコード部分

import re

def atoi(text):
    return int(text) if text.isdigit() else text

def natural_keys(text):
    return [ atoi(c) for c in re.split(r'(\d+)', text) ]

sorted(l, key=natural_keys)

このコードで私のやりたかったソートが実現できました。natural_keys()の中で呼ばれるatoi()部分は他で使わないことから、これをラムダ式にできると思い、1つにまとめてみました。少なくとも私の使い方だと同じ結果になっています。

おわりに

可読性については様々な議論があると思います。この例ではソートのキーに指定できるライブラリ的なものとしてラムダ式にしてみました。ライブラリをimportする場合、関数の中身は見ないですよね。それと同じ使い方で、必要なソースに「目をつぶってコピペして使える1行」になりました。
ソートの動作については、親文字列の中から正規表現で「連続した数字」を見つけて分解し、子文字列と数値の混合リストにしてソートするのですが、詳しくは参考にした記事をご覧ください。