python数组 *** 作容易犯的错误

在下面的例子中，希望做到的是数组a,b中元素大于3的置为1，其余为0。应该先做置0 *** 作。

程序：

import numpy as np

a = [1,2,2,3,4,5]
a = np.array(a)
b = [1,2,2,3,4,5]
b = np.array(b)

# 错误的例子
a[a>3] = 1
a[a<=3] = 0
print('a',a)

b[b<=3] = 0
b[b>3] = 1
print('b',b)

输出：

a [0 0 0 0 0 0]
b [0 0 0 0 1 1]