Q&A
Python
1import numpy as np 2 3a=np.array([[0, 0, 1, 0], 4 [0, 1, 0, 0], 5 [0, 0, 1, 0], 6 [0, 1, 0, 0], 7 [0, 0, 1, 0]])
このような4列x5行の行列aをnumpyで作成します。
ここで、3列目が1の行のインデックス(つまり0,2,4)
あるいは2列目が1の行のインデックス(つまり1,3)
を表示するにはどうしたら良いのでしょうか？
回答1件
0
ベストアンサー
python
1print(np.where(a[:,2]==1)[0]) 2# [0 2 4] 3 4print(np.where(a[:,1]==1)[0]) 5# [1 3]
投稿2022/11/25 03:35
総合スコア21265
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。