Python リストの並び替えについて

Question

こんにちは、
例えばゲームでインデックス0番目、1番目、2番目の人が勝利した回数をリストに[0,2,1]と書くとします。勝利した回数が1番多い人がランキングの0番目、2番目に多い人が1番目、1番少なかった人がランキングの下位である2番目になるようなリストを作りたいですが、その際、昇順や降順の順でなく、元のリストのポジションにランキングを対応させてリストを作りたいです。
この場合だと、元リスト（勝利した回数）=[0,2,1]で作りたいランキングリストは、index(0)は勝利した回数が1番少ないから=2,index(1)は勝利した回数が1番多いから=0 そして2番目に勝利回数が多いのはindex(2)なので、=1
になり作りたいリストは[2,0,1]です。
このとき、勝利した回数が同じであれば元のリストの順になります。例) [1,1,0]→[0,1,2]
sortedメソッドや argsort について調べて見ましたが、やはり自分が作りたいリストにならないです。
きちんと説明できてないかもしれないですが、アドバイスお願いします。

Answer 1

やりたいことは、argsortを二回やればできそうです。
numpyのargsortで、reverseかつstableなのはちょっと面倒そうなので、リストでやってます。
(もっといい方法がありそうな気もします)

a = [0, 2, 1]
b = sorted(range(len(a)), key=lambda x: a[x], reverse=True)
c = sorted(range(len(b)), key=lambda x: b[x])
print(c)
#[2, 0, 1]

a = [1, 1, 0]
b = sorted(range(len(a)), key=lambda x: a[x], reverse=True)
c = sorted(range(len(b)), key=lambda x: b[x])
print(c)
#[0, 1, 2]

numpyを使うなら、-1をかけてargsortすれば降順にソートできるので、シンプルに下記にできそうです。

import numpy as np

a = np.array([0, 2, 1])
b = np.argsort(np.argsort(-a, kind='stable'))
print(b)
#[2 0 1]

Comments

[bsdfan]

二回ソートするよりも、can110さんの回答のように、後半の処理は代入でやるほうが、数が多い時の効率は良さそうです。
参考: https://stackoverflow.com/questions/28573545, https://stackoverflow.com/questions/5284646

Answer 2

まずは元リストのインデックスを保持したまま値でソートします。
次に、欲しい結果が[0,2,1]の場合[2,0,1]で、[1,1,0]の場合[0,1,2]とのことなので
同じ大きさのリストを用意して、ソートされたインデックスの位置に連番をセットしていけばよいです。
numpy.argsortを使えば少し簡潔に書けます。

for lst in [[0,2,1],[1,1,0]]:
    idxs = [p[0] for p in sorted(enumerate(lst), key=lambda e:(e[1],-e[0]), reverse=True)] # 方法１
    #idxs = np.argsort(np.array(lst)*-1, kind='stable') # 方法２

    ret = [0 for _ in range(len(lst))]
    for i,pos in enumerate(idxs):
        ret[pos] = i
    print(lst,ret)

Answer 3

kyokio さんのアイデアでいいと思います。

li1 = [0,2,1]
li2 = [(i,li1[i]) for i in range(0, len(li1))]
li3 = sorted(li2, key=lambda x: x[1], reverse=True)
li4 = [y[0] for y in li3]
print(li4)

Answer 4

def get_rank(lst):
    ls = [(v,i) for i,v in enumerate(lst)]
    ls2 = [v for v in sorted(ls, key=lambda x:(x[0],-x[1]), reverse=True)]
    return [ls2.index(v) for v in ls]

wins = [0, 2, 1]       # 勝利した回数
rank = get_rank(wins)
print(rank)            # [2, 0, 1] ランキングリスト
wins = [1, 1, 0]
rank = get_rank(wins)
print(rank)            # [0, 1, 2]
wins = [1, 1, 2, 0, 2, 2, 1]
rank = get_rank(wins)
print(rank)            # [3, 4, 0, 6, 1, 2, 5]

Answer 5

辞書を使ってはどうでしょうか？

ranking_dict = {0:0, 1:2, 2:1}
print(ranking_dict)
# {0: 0, 1: 2, 2: 1}
ranking_list = sorted(ranking.items(), key=lambda x:x[1], reverse=True)
print(ranking_list)
# [(1, 2), (2, 1), (0, 0)]

Pythonの辞書の要素をソートする方法【初心者向け】

Answer 6

普段C++を書いているのでpythonで上手く書けないですが、リストにpairで[買った数、自分のインデックス]を持たせて買った数でsortして自分のインデックスの方を出力すればいいのではないでしょうか？

list = [[0, 0], [2, 1], [1, 2]]
ans = [x[1] for x in sorted(list, reverse=True)]
print(ans)
# [1, 2, 0]