多重配列の位置(index)を取得したい

Question

navigation([['Y', 0, 0, 0, 'C'],
             [ 0,  0, 0, 0,  0],
             [ 0,  0, 0, 0,  0],
             ['M', 0, 0, 0, 'S']]))

上記のような多次元配列があった時に
要素を選択すれば，その位置を返してくれるようなメソッドや方法があれば教えていただきたいです
リストのindex()メソッドのようなイメージです

以下のようなことが出来れば良いなと考えています．
よろしくお願いいたします

>>> print(〜('Y'))
(0, 1) # このような形でなくても大丈夫です

Comments

[hayataka2049]

0を指定するとどの位置を返すのでしょうか。0に限らず複数ある場合は？　あとはそもそもその配列をどうやって使う想定ですか？　navigationなるものに渡していますが、これはどういう型（クラス）ですか？

[退会済みユーザー]

Yをスタート地点として，C, M, Sそれぞれの距離を求めるために使う想定です.
上下・左右・斜めに進めるものとして，例えば投稿した配列考えますと
Y - C：4
Y - M：3
Y - S：4（斜め右下に3, 右に1）
ようなイメージです
ですので，0の指定や複数ある場合は想定せずで大丈夫です．

要素の位置がわかればこのような距離の計算もできると考え質問させていただきました
説明不足で申し訳ありません．

Answer 1

多次元配列である値のインデックスを得たい場合は numpy.where をお使いください。

import numpy as np

arr = np.array([['Y', 0, 0, 0, 'C'],
                [0,  0, 1, 1,  0],
                [0,  0, 1, 1,  0],
                ['M', 0, 0, 0, 'S']])

rows, cols = np.where(arr=='C')
# 返り値は各 axis ごとのインデックスに分かれている。
# (axis=0 のインデックスのリスト、axis=1 のインデックスのリスト)
print(rows, cols)  # [0] [4]

# (i, j) という形式で得たい場合は column_stack() を使う
indices = np.column_stack([rows, cols])
print(indices)  # [[0 4]]

rows, cols = np.where(arr=='1')
print(rows, cols)  # [1 1 2 2] [2 3 2 3]
indices = np.column_stack([rows, cols])
print(indices)
# [[1 2]
#  [1 3]
#  [2 2]
#  [2 3]]

Comments

[退会済みユーザー]

numpyですね！
ありがとうございます．とても参考になりました！