二重配列のインデックス取得

Question

二重配列のインデックス取得方法を教えてください。
例えば

arr = ["A","B","C","D"]
print(arr.index("D"))

とやると、期待通り３と表示されますが、

arr = [["A","B","C"],["D","E","F"],["G","H","I"]]

のとき、print(arr.index("G"))では当然エラーになります。
期待する"G"のインデックス[2][0]を取得するには、どうすればいいでしょうか？
もしくはindex()メソッド以外の方法でも構いません。
よろしくお願いします。

Answer 1

解決済みではありますが。

import numpy as np
arr = [["A","B","C"],["D","E","F"],["G","H","I"]]
arr = np.array(arr)
print(np.argwhere(arr=="G"))
# [[2 0]]

Answer 2

素直な方法。

arr = [["A", "B", "C"], ["D", "E", "F"], ["G", "H", "I"]]

for y, row in enumerate(arr):
    try:
        pos = (y, row.index("G"))
        break
    except ValueError:
        pass

print(pos)

ちょっと捻った方法。

from itertools import chain

arr = [["A", "B", "C"], ["D", "E", "F"], ["G", "H", "I"]]

pos = divmod(
    list(chain(*arr)).index("G"), len(arr[0])
)
print(pos)

内包表記を使った方法。

arr = [["A", "B", "C"], ["D", "E", "F"], ["G", "H", "I"]]

pos = [
    (x, y) for y, row in enumerate(arr) for x, elem in enumerate(row) if arr[x][y] == "G"
][0]
print(pos)

NumPyを使うのもアリかもしれませんね。

import numpy as np
arr = np.array(
    [["A", "B", "C"], ["D", "E", "F"], ["G", "H", "I"]]
)

pos = tuple(nd[0] for nd in np.where(arr == "G"))
print(pos)

Comments

[退会済みユーザー]

さっそく４つも解法を提案していただきありがとうございます！
勉強になります

[LouiS0616]

『必ず検索に成功すること』が前提となっていることだけご注意を。

Answer 3

多重かつ要素数が異なる(jagged)リストで考えてみました。

# srcリストからfind_valに一致する要素位置をidx_findリストに格納して返す
def search( src, find_val, idx_find, idx_cur = []):
    if isinstance( src, list): # リストのみで構成されていることが前提
        for i, v in enumerate(src):
            search( v, find_val, idx_find, idx_cur + [i])
    else:
        if src == find_val:
            idx_find.append( idx_cur)

def test(  arr):
    idx_find = []
    search( arr, 'G', idx_find)
    print(idx_find)

# １つ合致
test( [['A','B','C'],['D','E','F'],['G','H','I']] )    # [[2, 0]]
test( ['A',['B',['C','D',['E','F']],'G'],['H','I']] )  # [[1, 2]]

# ２つ合致
test( [['G','B','C'],['D','E','F'],['G','H','I']] )    # [[0, 0], [2, 0]]
test( ['G',['B',['C','D',['E','F']],'G'],['H','I']] )  # [[0], [1, 2]]