python numpy创建逻辑数组

我有一个大的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)

我想要

a

中所有

2.0

s的

1.0

（浮点/双精度）

a==2.0

给出了

bool

数组，但不是

1.0

浮点数组。我怎么做呢？

试试看

(a==2.0) * 1

True乘以数字等于该数字，False乘以数字等于0（a==2）。aType（浮点） 数组（[1,1,1,0,0,0,0,0,1.]）） 只需执行

a=float（a）

，您还可以使用

numpy

转换类型

[1.0 if i else False for i in r]

---

您希望非

2.0

元素是什么<代码>错误<代码>0.0？它们在

a

中的值？此外，您还应注意在不考虑