python numpy数组中的复制问题

Posted 2020-10-07

vector = numpy.array([5, 10, 15, 20])
equal_to_ten_or_five = (vector == 10) | (vector == 5)
vector[equal_to_ten_or_five] = 50
print(vector)

　

第一次看到这个的时候一脸懵逼，后来分析了下懂了下面记录下，方便下次看看

第一行分析:
结果5, 10, 15, 20

第二行分析:
vector == 10 数组和值比对获得结果是每个元素和这个数比较生成相应的bool数组 这里的话就知道了结果是[Flase,True,Flase,Flase]
vector == 5就是[True,Flase,Flase,Flase]
然后进行或‘|‘运算得到了[True,True,Flase,Flase]
着两个数组进行或运算的时候是数组直接相同序号的元素或元素然后得到新的数组(遵循规则F|F =F,F|T=T,T|T=T)
此时equal_to_ten_or_five 就等于[True,True,Flase,Flase]
第三行分析:
vector[equal_to_ten_or_five]=50 会给等于True的复制为50 其他的不变也就是此时
vector等于[50,50,15,20]