フィールド名のインデックスを覚える必要なく、名前付きタプルのリストをソートしたい。私の解決策はかなり厄介なようで、誰かがよりエレガントな解決策を持つことを望んでいました。
from operator import itemgetter
from collections import namedtuple
Person = namedtuple('Person', 'name age score')
seq = [
Person(name='nick', age=23, score=100),
Person(name='bob', age=25, score=200),
]
# sort list by name
print(sorted(seq, key=itemgetter(Person._fields.index('name'))))
# sort list by age
print(sorted(seq, key=itemgetter(Person._fields.index('age'))))
ありがとう、ニック
from operator import attrgetter
from collections import namedtuple
Person = namedtuple('Person', 'name age score')
seq = [Person(name='nick', age=23, score=100),
Person(name='bob', age=25, score=200)]
リストを名前で並べ替える
sorted(seq, key=attrgetter('name'))
年齢でリストを並べ替える
sorted(seq, key=attrgetter('age'))
sorted(seq, key=lambda x: x.name)
sorted(seq, key=lambda x: x.age)
@zenpoyはパフォーマンスを懸念していたため、ここで示した2つの選択肢の速度をテストしました。
テストスクリプト:
import random
from collections import namedtuple
from timeit import timeit
from operator import attrgetter
runs = 10000
size = 10000
random.seed = 42
Person = namedtuple('Person', 'name,age')
seq = [Person(str(random.randint(0, 10 ** 10)), random.randint(0, 100)) for _ in range(size)]
def attrgetter_test_name():
return sorted(seq.copy(), key=attrgetter('name'))
def attrgetter_test_age():
return sorted(seq.copy(), key=attrgetter('age'))
def lambda_test_name():
return sorted(seq.copy(), key=lambda x: x.name)
def lambda_test_age():
return sorted(seq.copy(), key=lambda x: x.age)
print('attrgetter_test_name', timeit(stmt=attrgetter_test_name, number=runs))
print('attrgetter_test_age', timeit(stmt=attrgetter_test_age, number=runs))
print('lambda_test_name', timeit(stmt=lambda_test_name, number=runs))
print('lambda_test_age', timeit(stmt=lambda_test_age, number=runs))
結果:
attrgetter_test_name 44.26793992166096
attrgetter_test_age 31.98247099677627
lambda_test_name 47.97959511074551
lambda_test_age 35.69356267603864
ラムダの使用は確かに遅くなりました。最大10%遅くなります。
[〜#〜] edit [〜#〜]:
さらにテストすると、複数の属性を使用してソートした場合の結果が示されます。同じセットアップで次の2つのテストケースを追加しました。
def attrgetter_test_both():
return sorted(seq.copy(), key=attrgetter('age', 'name'))
def lambda_test_both():
return sorted(seq.copy(), key=lambda x: (x.age, x.name))
print('attrgetter_test_both', timeit(stmt=attrgetter_test_both, number=runs))
print('lambda_test_both', timeit(stmt=lambda_test_both, number=runs))
結果:
attrgetter_test_both 92.80101586919373
lambda_test_both 96.85089983147456
ラムダのパフォーマンスは依然として劣っていますが、それほど劣っていません。約5%遅くなりました。
テストはPython 3.6.0。
itemgetter()の使用について誰も言及していないため、ここではitemgetter()の使用方法を説明します。
from operator import itemgetter
from collections import namedtuple
Person = namedtuple('Person', 'name age score')
seq = [
Person(name='nick', age=23, score=100),
Person(name='bob', age=25, score=200),
]
# sort list by name
print(sorted(seq, key=itemgetter(0)))
# sort list by age
print(sorted(seq, key=itemgetter(1)))
これは一部の人にとっては少し「魔法的」かもしれませんが、私は次のように部分的です:
# sort list by name
print(sorted(seq, key=Person.name.fget))