Q&A
pythonで同じ要素数の２つの配列があり，それらのうちの一つの配列をソートする際，その並びかえと同じようにもう一つの配列も並びかえる高速な方法はありますか？
python
1import numpy as np 2 3a = np.array([5, 1, 3, 6, 7, 2]) 4b = np.array([6, 4, 7, 8, 5, 8])
という配列a,bがそれぞれあるときに
aをソートした結果
a = [1, 2, 3, 5, 6, 7]
とaの並びかえ順をbに適用した配列
b = [4, 8, 7, 6, 8, 5]
を高速に得られる方法が知りたいです．
お願いいたします．
回答1件
0
ベストアンサー
numpy.argsortが使えます。
値でソートした結果として配列インデックスを返します。
Python
1import numpy as np 2 3a = np.array([5, 1, 3, 6, 7, 2]) 4b = np.array([6, 4, 7, 8, 5, 8]) 5 6idx = a.argsort() 7print(a[idx]) # [1 2 3 5 6 7] 8print(b[idx]) # [4 8 7 6 8 5]
投稿2021/10/01 07:57
総合スコア38341
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。