python numpy creating logical array

Question

I have large numpy array and trying to create logical (1, 0) array from it. For example,

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)

I want 1.0 (float/double) for all 2.0 s in a . a==2.0 gives array of bool , but not 1.0 float. How can I do it?

Answer 1

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

Answer 2

Try

(a==2.0) * 1

True multiplied by number will be equal to that number and False multiplied by number will be equal to 0

Answer 3

只需执行a = float(a) ，您也可以使用numpy来转换类型

Answer 4

你可以试试这个

[1.0 if i else False for i in r]