一次元配列と二次元配列の関連付け

前提・実現したいこと

ここに質問の内容を詳しく書いてください。
一次元配列と二次元配列を関連付けたいのですがどのようにすればいいかわからない状態です

発生している問題

一次元配列が
ary = array([0.42317162, 0.05656336, 0.02925958, 0.00797409, 0.00162188,
       0.01255612, 0.00166179, 0.00578814, 0.00216196, 0.00955845,
       0.01244116, 0.00758913, 0.00758913, 0.01244116, 0.00955845,
       0.00216196, 0.00578814, 0.00166179, 0.01255612, 0.00162188,
       0.00797409, 0.02925958, 0.05656336, 0.42317162])
二次元配列が
min_id = (array([    5,  2425,  3561,  5993,  7899,  9488, 10576, 12057, 13216,
        15837, 17784, 19832, 24268, 26316, 28263, 30884, 32043, 33524,
        34612, 36201, 38107, 40539, 41675, 44095]),)
のとき二次元配列の１つ目の要素(min_id[0]の要素)とaryを関連付けて並べ替えたいです。
ary[0]とmin_id[0][0]、ary[1]とmin_id[0][1]のように対応させて、
その後にaryを昇順にしたときmin_idも先程対応させたaryに対応するように順番を変えたいです

(ary[1],ary[0],ary[2])という順番になったとき(min_id[0][1],min_id[0][0],min_id[0][2])という順番にしたいです

試したこと

辞書などを使えば行けるのかなと思ったのですが使い方がいまいちわからず、できませんでした、、、

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

回答1件

ベストアンサー

回答

ary が形状が (24,) の numpy 配列、min_id が形状が (1, 24) の numpy 配列としたとき、ary を昇順ソートした場合、それに連動して min_id を並び替えるコードは以下になります。

numpy.argsort で並び替え後のインデックスの値が取得できるので、それを利用して、numpy の Fancy Indexing を使います。

サンプルコード

python
1import numpy as np
2
3ary = np.array(
4    [0.42317162, 0.05656336, 0.02925958, 0.00797409, 0.00162188, 0.01255612, 0.00166179, 0.00578814, 0.00216196, 0.00955845, 0.01244116, 0.00758913, 0.00758913, 0.01244116, 0.00955845, 0.00216196, 0.00578814, 0.00166179, 0.01255612, 0.00162188, 0.00797409, 0.02925958, 0.05656336, 0.42317162]
5)
6min_id = np.array(
7    [[5, 2425, 3561, 5993, 7899, 9488, 10576, 12057, 13216, 15837, 17784, 19832, 24268, 26316, 28263, 30884, 32043, 33524, 34612, 36201, 38107, 40539, 41675, 44095]]
8)
9
10print(min_id[0, ary.argsort()])  # ary の値を昇順ソートした場合、それに対応する min_id を並び替える。
11
12# [36201  7899 33524 10576 13216 30884 32043 12057 19832 24268  5993 38107
13#  15837 28263 17784 26316  9488 34612  3561 40539 41675  2425     5 44095]