行列の要素の位置？？？

Question

このbの行列の0の要素の位置を調べようと思ったのですが正しい結果が出ない為質問しました。

本来、結果としては
array[0,3],array[0,1]
コード
b = np.array([[1],
[0],
[4],
[0]])

print(b==0)

z = np.where(b == 0)
print(z)

結果
[[False]
[ True]
[False]
[ True]]

(array([1, 3]), array([0, 0]))

Answer 1

この場合、numpy.where() が返すのは (0がある行のインデックス一覧, 0がある列のインデックス一覧) です。[(行, 列), (行, 列), ...] というリストを返すわけではありません。

import numpy as np

a = np.array([[1],
              [0],
              [0],
              [4],
              [0]])

rows, cols = np.where(a == 0)
print(rows)  # [1 2 4]
print(cols)  # [0 0 0]

for row, col in zip(rows, cols):
    print((row, col))
# (1, 0)
# (2, 0)
# (4, 0)

Comments

[tamura_ryoma]

非常にわかりやすいご説明
ありがとうございました。
助かりました。