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]