python numpy創建邏輯數組

Question

我有一個很大的 numpy 數組，並試圖從中創建邏輯 (1, 0) 數組。 例如，

a=np.array((2.0,2.0,2.0,3.0,4.0,5.0,6.0,2.0))
a==2.0
array([ True,  True,  True, False, False, False, False,  True], dtype=bool)

我想1.0 （浮動/雙），用於所有2.0以s a 。 a==2.0給出bool數組，但不是1.0 float。 我該怎么做？

Answer 1

>>> (a == 2).astype(float)
array([1., 1., 1., 0., 0., 0., 0., 1.])

Answer 2

嘗試

(a==2.0) * 1

真乘以數字將等於該數字，假乘以數字將等於 0

Answer 3

只需執行a = float(a) ，您也可以使用numpy來轉換類型

Answer 4

你可以試試這個

[1.0 if i else False for i in r]