Pythonで文字と数字が混ざった2次元配列のソート

Question

Pythonでcsvから取得した２次元配列のリストを２行目でソートをしたかったのですが、データにfloatと文字が混じってしまうため、sortedのitemgetterでソートしようとすると文字列としてソートされるのですが、数字順でソートして最後に文字が来るようなソートは出来ないのでしょうか？

#元データ
list=[
[1, 10.1],
[2, 1.1],
[3, 2.2],
[4, "無"]
]
rows = [[row[0], row[1]] for row in list]
result = sorted(rows, key=operator.itemgetter(1))
print(result)
#[2, 1.1],
#[1, 10.1],
#[3, 2.2],
#[4, 無]

#期待する結果
list=[
[2, 1.1],
[3, 2.2],
[1, 10.1],
[4, 無]
]

Comments

[A_kirisaki]

そもそも float と str の比較無理やんけって怒られません？

[t_obara]

keyに関数とか指定できるので、その中でいいように実装すれば良いかと。

Answer 1

keyを指定してください。
さらに、keyで指定する関数で浮動小数以外の場合は無限大を返すようにしておきましょう。

>>> def keyfunc(x):
...     if type(x[1]) == float:
...         return x[1]
...     else:
...         return float('inf')
...
>>> list=[
...       [1, 10.1],
...       [2, 1.1],
...       [3, 2.2],
...       [4, "無"]
...      ]
>>>
>>> print(sorted(list, key=keyfunc))
[[2, 1.1], [3, 2.2], [1, 10.1], [4, '無']]

Comments

[enigumalu]

ありがとうございます、標準機能になさそうだったのでkey作るしかなさそうと思ってたので参考にさせていただきました！