Python3 多次元リストの特定要素のインデックスの取得

Question

一次元のリストで、特定の要素のインデックスを取得するときにli.index()を使っています。

二次元にしたときに、使えません。
index([*, ])などと指定するはずですが、内側のリストの並びが不明だからです。
例えば "bbb" を検索するときに、[, "bbb"] なのか ["bbb", ] なのかなど、一次元のときの並びがわかりません。
当然、リストの要素数もわかりません。[, ] なのか [, *, *] なのか。

forを使ってインデックスを返す関数を自作すればできないことは無いんですが、他にいい方法とかあったら教えてほしいです。「いい方法」とは既存のライブラリを使う、自作関数でもっと効率よく検索する等。
例えばindex()でワイルドカードを指定する方法など。

# 二次元リストのインデックス取得関数
def get_index(li, data):
    index = []
    for i, a in enumerate(li):
        for j, b in enumerate(a):
            if data == b:
                index = [i, j]
    return index

if __name__ == "__main__":
    li = [
        ["aaa", "bbb", "ccc"],
        ["ddd", "eee", "fff"],
        ["ggg", "hhh", "iii"]
    ]

    index = get_index(li, "bbb") # "bbb" を検索
    print(index)

Answer 1

普通の多次元リストならnumpy.argwhereを使うと楽です。
各要素数、階層が異なるJagged arrayも含めて考えると、以下のような再帰処理で求めることができます。

import numpy as np

# n次元リスト版
def get_idxs(obj, data, idxs, cur_idx = []):

    # 先頭だけでよければ以下を生かす
    #if len(idxs) >= 1:
    #    return

    if isinstance(obj, list):
        for i, a in enumerate(obj):
            get_idxs(a, data, idxs, cur_idx + [i])
    elif obj == data:
        idxs.append(cur_idx)

li = [
    ["aaa", "bbb", "ccc"],
    ["ddd", "eee", "fff"],
    ["ggg", "hhh", "iii"]
]

a = np.array(li)
index = np.argwhere(a == 'bbb')
print(index) # [[0 1]]

idxs = []
get_idxs(li, "bbb", idxs)
print(idxs) # [[0, 1]]

li = [ ['a','x'],
       ['a','b','x',
          ['a','x','c']],
        'x',
        'c']

idxs = []
get_idxs(li, 'x', idxs)
print(idxs) # [[0, 1], [1, 2], [1, 3, 1], [2]]

Comments

[person]

ありがとうございます。
試してみます。