python ３次元配列 １～ｎ番目に大きいインデックスを反す

Question

pythonの配列表示でインデックスを出力する方法。

３次元配列で、最も大きい値～ｎ番目に大きい値を持つインデックスを反すプログラムを組みたいです。
なにか手がありましたら教えて頂けないでしょうか。

    list = [[[0 for i in range(256)] for j in range(256)]for k in range(256)]
    
   
    for y in range(height):
        for x in range(width):
            
            r, g, b = img[x,y]
            print(r,g,b)
            
            list[r][g][b] = list[r][g][b] + 1
    
   list.sort(reverse=True)

Answer 1

こんなふうにできそうです。

indice = np.unravel_index(
    np.argsort(arr, axis=None)[::-1], arr.shape
)
indice = np.transpose(indice)

NumPy苦手なので遠回りなことをしているような気もしますが。

Answer 2

以下のような感じでどうでしょうか。

import numpy as np

# 元データ
l = [[[1,2,3],[4,5,6]]]
a = np.array(l)

# インデックス配列を得る
b = np.where(a==a)

# インデックスと値を１要素として持つリストを作成
lst = [(pos , a[pos]) for pos in zip(*b)]

# 値で降順ソート
lst.sort( key=lambda v: v[1], reverse=True)
print(lst) # [((0, 1, 2), 6), ((0, 1, 1), 5), ((0, 1, 0), 4), ((0, 0, 2), 3), ((0, 0, 1), 2), ((0, 0, 0), 1)]

print(lst[0][0]) # (0, 1, 2)

Answer 3

値と元のインデックスを保持するクラスを作って線形リストにし、値でソートして n 番目を返せばいいのではないかと。