Python3.6.2を使用しております。
2次元配列で特定の列を検索し、その値をもつ行の番号と数を得たいです。
例えば
list=[[0 2 0 0 0 0]
[5 1 0 0 0 0]
[2 1 0 0 0 0]
[7 1 0 0 0 0]]
という配列があり2列目に1がある行の
番号を[1 2 3]、数を3という形で返してほしいのです。
試したには試したのですが、
numpy.where(list[:,1]==1)
len(numpy.where(list[:,1]==1))
(array([1, 2, 3], dtype=int64),)
1
このように上手くいきません。
補足
numpy.where(list[:,1]==1)[0]
[1 2 3]
でなんとかなるのですが、これは二次元配列の中身ですよね。
一次元配列を得る方法を教えていただけないでしょうか。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2017/09/10 11:35