3次元配列のソート方法

Question

###実現したいこと
3次元配列をリストwaruの値のついてソートしたいです。
めざしている実行結果は

waru=[3.0, 4.0, 6.0, 5.0, 2.0]
ijkari=[[10, 21, 5, 30, 5], [30, 84, 30, 150, 10], [3.0, 4.0, 6.0, 5.0, 2.0]] 
ij=[[5, 10, 21, 30, 5], [10, 30, 84, 150, 30], [2.0, 3.0, 4.0, 5.0, 6.0]]

waruの中身が現在[3.0, 4.0, 6.0, 5.0, 2.0]なので同時にリストijkariの値を同時に並べ替えたい。

|pi|10|21|5|30|5|
|:--|:--:|--:|
|wi|30|84|30|150|10|
|waru|3|4|6|5|2|

から
|pi|5|10|21|30|5|
|:--|:--:|--:|
|wi|10|30|84|150|30|
|waru|2|3|4|5|6|

一番下のwaruに応じてpi,wiも並びが変わっているとういうような並べ替えをしたいです。

###問題点
waruの値でソートされていない。
求めている実行結果にする方法を教えてください。

5 10 21 30 5
10 30 84 150 30
waru= [3.0, 4.0, 6.0, 5.0, 2.0]
ijkari= [[10, 21, 5, 30, 5], [30, 84, 30, 150, 10], [3.0, 4.0, 6.0, 5.0, 2.0]]     
ij= [[10, 21, 5, 30, 5], [3.0, 4.0, 6.0, 5.0, 2.0], [30, 84, 30, 150, 10]]

###コード

from operator import itemgetter

pi=[int(i) for i in input().split()]
wi=[int(j) for j in input().split()]
pin=len(pi)
win=len(wi)

waru=[]
for i in range(pin):
    w=wi[i]/pi[i]
    waru.append(w)
print("waru=",waru) #確かめ
#pi,wiをwi/piについてソートする(3次元配列)
ijkari=[]
ijkari=[pi,wi,waru]
print("ijkari=",ijkari)
ij=sorted(ijkari,key=itemgetter(2))
print("ij=",ij)

Comments

[meg_]

「waruの値のついてソートしたい」が分かりません。具体的な方法は何でしょうか？

[meg_]

> waru=[3.0, 4.0, 6.0, 5.0, 2.0]
> waru 3 4 6 5 0（ソート前）
> waru 3 4 5 6 7（ソート後）

リストwaruの中身が変わってしまっているのですが、タイプミスでしょうか？
リストwaru自体は昇順の並び替えをしたい、ということで良いですか？

[langhtorn]

そうです。タイプミスです。ご指摘ありがとうございます。

Answer 1

下記処理ではijはfloatのnumpy配列となりますので、必要に応じてリストに変換する等処理してください。

import numpy as np

waru=[3.0, 4.0, 6.0, 5.0, 2.0]
ijkari= [[10, 21, 5, 30, 5], [30, 84, 30, 150, 10], [3.0, 4.0, 6.0, 5.0, 2.0]]     

ij = np.array(ijkari)[:, np.argsort(waru)]
print(ij)
#array([[  5.,  10.,  21.,  30.,   5.],
#       [ 10.,  30.,  84., 150.,  30.],
#       [  2.,   3.,   4.,   5.,   6.]])

Comments

[langhtorn]

ありがとうございます。参考にさせていただきます。

Answer 2

入力
waru=[3.0, 4.0, 6.0, 5.0, 2.0]
ijkari=[[10, 21, 5, 30, 5], [30, 84, 30, 150, 10], [3.0, 4.0, 6.0, 5.0, 2.0]]
出力
ij=[[5, 10, 21, 30, 5], [10, 30, 84, 150, 30], [2.0, 3.0, 4.0, 5.0, 6.0]]
の理解で、そして ijkari の一要素が waru の前提でのコードです。

waru= [3.0, 4.0, 6.0, 5.0, 2.0]
ijkari= [[10, 21, 5, 30, 5], [30, 84, 30, 150, 10], [3.0, 4.0, 6.0, 5.0, 2.0]]

def index_sort(slst, klst):
    try:
        idx = slst.index(klst)
    except ValueError as e:
        print(f'ValueError: {e}')
        return
    tmp = list(map(list, zip(*slst)))
    tmp = sorted(tmp, key=lambda x:x[idx])
    return list(map(list, zip(*tmp)))

ij = index_sort(ijkari, waru)
print(ij) # [[5, 10, 21, 30, 5], [10, 30, 84, 150, 30], [2.0, 3.0, 4.0, 5.0, 6.0]]

Comments

[langhtorn]

ありがとうございます。参考にさせていただきます。