根据另一个np.array python的值设置np.array的值

I have a numpy array A, which contains values between 0 and 1. I want to create another numpy array y, such that the value of y(i) = 1 if A(i) >= 0.5, and y(i) = 0 if A(i) < 0.5. I used the following python code: I have a numpy array A, which contains values b