へっぽこエンジニア@機械学習

へっぽこなエンジニアがへっぽこなりに機械学習の習得に取り組む過程で得た知識について書いたブログです。実装に際してハマった点や、基礎的な内容についての記載が中心ですが、徐々に高度化させて行ければと思っています。非へっぽこの方からは間違いの指摘など頂ければ嬉しいです。

ndarrayで条件にあう要素のみの和(加算)を求める際の注意点

ndarrayの中で特定の条件にあう要素のみの和を求めたいとき、

例えば以下のndarrayを考える。

f:id:growingEngineer:20190314093146p:plain

例えば0以上の要素の和を求める時、正解は'55'だが以下の様な誤ったケースがある。

f:id:growingEngineer:20190314093410p:plain

上記では和ではなく、条件を満たす要素数をカウントしてしまっている。

それは以下の挙動から分かると思う。

f:id:growingEngineer:20190314093526p:plain

カッコ内はTrue/Falseを出力するので、それをsumするとTrue(=1)を足してしまうのだ。

(間違ってたらごめんなさい)

 

正しくは以下の通り。

f:id:growingEngineer:20190314093726p:plain