np.where()[0][0]←この[0][0]の意味が分からない

Question

np.where()[0][0]←この[0][0]の意味が分からない

python 3.9.7 を使用しております。

import numpy as np 

array = np.array([["a","b","c"],["d","e","f"],[1,2,3]])

print(array[np.where(array == "d")[0][0]][:])
print(np.where(array == "d")[0][0])

の

print(array[np.where(array == "d")[0][0]][:])
print(np.where(array == "d")[0][0])

の部分の[0][0]の意味が分かりません。

ちなみに先ほどのコードを実行すると、

['d' 'e' 'f']
1

と出力されます。どなたかご教授願います。

Answer 1

回答出てますが、違う観点から。

np.where(array == "d")[0][0]

これは、

tmp = np.where(array == "d")
tmp[0][0]

と同じです。
tmpは.where()返り値を保持しているのですが、tmpの使いどころが2行目の結果だけであれば代入しなくても、それをそのまま使えばいいということです。、

s = 'data'
s.upper()

'data'.upper()

も同じですね。

Comments

[eri555]

なるほど！より理解が深まりました！ありがとうございます。

Answer 2

array == "d"の結果は

array([[False,False,False],
       [True,False,False],
       [False,False,False]

となります。これはarrayの2行目1つ目の要素が"d"だからです。

そして、np.where(array == "d")の結果は(array([1]),array([0]))です。

このタプルから[0]で1つ目の要素のarray([1])を取り出して次の[0]でそのarrayから要素の1を取り出しています。

ですのでnp.where(array == "d")[0][0]は1になります。つまりarray[1]になります。これはarray(['d' 'e' 'f'])です。

Comments

[eri555]

とても分かりやすく教えていただき、ありがとうございます。
ちゃんと理解することができました。