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 (float / double) для всех 2.0 с в 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

Вы можете попробовать это

[1.0 if i else False for i in r]

Answer 3

Просто выполните a = float(a), вы также можете использовать numpy для преобразования типа

Answer 4

Try

(a==2.0) * 1

Значение True, умноженное на число, будет равно этому числу, а значение False, умноженное на число, будет равно 0