かべぎわブログ

ブログです

Pythonで2次元配列をソートする

Pythonで2次元配列をソートする場合どのようにしたほうがいいのかをご紹介します。

単一のキーでソートする

たとえばこんなかんじの2次元配列があったとする。

wawawa_array = [[7,5,1],[2,4,9],[7,4,9]]

こんなかんじにすると各配列の1番目の値でソートできる

wawawa_array = sorted(wawawa_array,key=lambda x:(x[0]))

ソートされた結果はこんなかんじ

>>> print(wawawa_array)
[[2, 4, 9], [7, 5, 1], [7, 4, 9]]



複数のキーでソートする

たとえばこんなかんじの2次元配列があったとする。

wawawa_array = [[7,5,1],[2,4,9],[7,4,9]]

こんなかんじにすると各配列の1番目の値が同じ値だったときに2番めの値でソートできる

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

ソートされた結果はこんなかんじ

>>> print(wawawa_array)
[[2, 4, 9], [7, 4, 9], [7, 5, 1]]