【python】インデックスに配列を用いた場合の、配列の各要素へのアクセスについて

Question

下記のprintの結果が「#想定していた結果↓」のようにならない理由をお伺いしたいです。

e[np.array([0,2])…

上記の[]が表しているのは、
「配列eの中の２次元配列のうち、０番目の配列にある２番目の要素」
という意味と考えていたところ、「#出力結果」のようになりました。
よろしくお願いします。

import numpy as np

e = np.zeros((3,3))
f = np.array([8,9])

e[np.array([0,2]), np.array([0,1])] = f
print(e)
#出力結果
[[8. 0. 0.]
 [0. 0. 0.]
 [0. 9. 0.]]

# 想定していた出力結果↓
[[0. 9. 8.]
 [0. 0. 0.
 [0. 0. 0.]]
コード

Answer 1

カンマで区切った場合はe[axis=0への指定, axis=1への指定]として処理されますので、代入先は0,0と2,1の位置です。

Comments

[dahachin]

ご回答ありがとうございます！
”e[axis=0への指定…”　回答に記載されている”axis”とはどういう
意味か教えていただくことは可能でしょうか？

[hayataka2049]

検索すると親切な説明が出てきますので、私からお教えすることはしません

Answer 2

e[np.array([0,2])…

は、「配列eの中の2次元配列のうち、0番目の配列と、2番目の配列」でしょう。

e[np.array([0,2]), np.array([0,1])] = f

は、

e[np.array([0,2])[0], np.array([0,1])[0]] = f[0]
e[np.array([0,2])[1], np.array([0,1])[1]] = f[1]

つまり、

e[0, 0] = 8
e[2, 1] = 9

Comments

[dahachin]

ご回答いただきありがとうございます。
”e[np.array([0,2]), np.array([0,1])] = f”
のうち、カンマの前が対象の配列、後が配列内の要素の番号（何番目の要素か）を
表しているという認識でよろしいでしょうか。

[otn]

そうですね。[ ]の使い方の基本です。

[dahachin]

よくわかりました、ありがとうございます！