pythonにて2次元配列の並び替え

[['acaa' 'xxx' 'vr301']
['abab' 'xxx' 'vr201']
['acac' 'xxx' 'vr101']]

このような配列を下記のように
vrの次の数字で昇順ソートしたいと考えています。

[['acac' 'xxx' 'vr101']
['abab' 'xxx' 'vr201']
['acaa' 'xxx' 'vr301']]

文字列でのソートの仕方が見つけられず困っております。
ご教授いただければ幸いです。よろしくお願いいたします。

行動規範の内容に同意します

回答2件

python
1lst = [
2  ['acaa', 'xxx', 'vr301'],
3  ['abab', 'xxx', 'vr201'],
4  ['acac', 'xxx', 'vr101'],
5  ['acac', 'xxx', 'vr55'],
6  ['acac', 'xxx', 'vr0'],
7]
8
9lst_sorted = sorted(lst, key=lambda x: int(x[2][2:]))
10
11from pprint import pprint
12pprint(lst_sorted)
13
14#
15[['acac', 'xxx', 'vr0'],
16 ['acac', 'xxx', 'vr55'],
17 ['acac', 'xxx', 'vr101'],
18 ['abab', 'xxx', 'vr201'],
19 ['acaa', 'xxx', 'vr301']]

投稿2021/12/07 10:55

melian

総合スコア19865

ベストアンサー

リストなら以下です。

python
1>>> print(data)
2[['acaa', 'xxx', 'vr301'], ['abab', 'xxx', 'vr201'], ['acac', 'xxx', 'vr101']]
3>>> print(sorted(data, key=lambda x: x[2]))
4[['acac', 'xxx', 'vr101'], ['abab', 'xxx', 'vr201'], ['acaa', 'xxx', 'vr301']]

numpyのndarrayなら以下です。

python
1>>> print(a)
2[['acaa' 'xxx' 'vr301']
3 ['abab' 'xxx' 'vr201']
4 ['acac' 'xxx' 'vr101']]
5>>> print(a[np.argsort(a[:,2])])
6[['acac' 'xxx' 'vr101']
7 ['abab' 'xxx' 'vr201']
8 ['acaa' 'xxx' 'vr301']]