###前提・実現したいこと
列ベクトルをいくつか配置した行列を扱っているのですが、
列ベクトルが全て0かどうかを調べて、全て0である列ベクトルの要素番号を全て取得したいです。
###例
python
1import numpy as np 2ary = np.array([[1, 0, 0], [0, 0, 0], [0, 0, 1], [0, 0, 0]]) 3ind = do_something(ary) 4assert ind == [1, 3]
###試したこと
python
1np.nonzero(ary)
だと、望むような結果は得られません。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
退会済みユーザー
2017/09/19 05:22