np.sumの使い方

Question

前提・実現したいこと

知人から頂いたコードでよく分からない内容のif文があったのでお伺いしたいです。

発生している問題

if np.sum(cm == np.array([0., 0., 0.]))==3:

試したこと

cmはある重心の座標で1*3のndarrayとなっております。
cm == np.array([0., 0., 0.])でしたらそのnp.sumを取ったものは0になるのではないのでしょうか?
np.sumの中身に==があるような表記法がわからず，調べても出てこなかったのでこちらにてお伺いします。

Answer 1

cm.shapeが(1,3)である場合、

cm == np.array([0., 0., 0.])

はNumpyのブロードキャストにより、cm == np.array([[0., 0., 0.]])と解釈されます。

その結果は(1,3)の形状を持つbool型の配列です。
bool型のTrueは整数の1、Falseは整数の0と解釈されますので、

np.sum(cm == np.array([0., 0., 0.]))==3

は、cm[0,0]、cm[0,1]、cm[0,1]が全て0.0であるという意味になります。

Comments

[Lily_1007]

bool型でTrueが1と解釈されているということなのですね。大変助かりました。誠にありがとうございます。