f(x) (0<=x<=1)を閾値によって０か１を返す方法について

Question

前提・実現したいこと

ここに質問の内容を詳しく書いてください。
pythonの分類に関する質問です。ある関数f(x) (0<= f(x)<= 1)をf(x) >= 0.5のとき1, f(x) < 0.5のとき0に分類したくて本に書いてあるサンプルコードを見たのですが、なぜこのコードで1か0を返すような関数になるのでしょうか。

該当のソースコード

def classify(x):
    return (f(x) >= 0.5).astype(np.int)

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

PythonではTrue/Falseがそれぞれ1/0に対応しているからです。

>>> int(True)
1
>>> int(False)
0

Comments

[hirotaka_i]

ありがとうございます！

Answer 2

f(x)はnumpyの配列を返す関数でいいですよね？

a = f(x) >= 0.5

これでaはbool型の配列です。それをintに変換してます。

例）

np.array([True, False, False])
array([ True, False, False])

a.astype(np.int)
array([1, 0, 0])

Comments

[hirotaka_i]

理解できました、ありがとうございます！