遇到一个numpy 错误

今天遇到了一个numpy错误。

import numpy

a=numpy.array(range(20))

a.shape=4,5

print a

array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19]])

#想让a的第1列大于0的全部归0.于是执行：

a[numpy.where(a[:,0]>0)][:,0]=0
>>> a
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19]])

#什么都没变啊啊啊啊，折腾了我快一个小时。

#原因应该是用where索引的时候返回的是一个原变量的临时副本，对副本进行修改，原来的变量不会改变。

#要解决这个问题，需要曲线救国：

temp_list=numpy.where(a[:,0]>0)[0]

a[temp_list,0]=0 #这样，a的第一列大于0的元素就会全部归零了~

#最终发现，原来可以这样~

a[numpy.where(a[:,0]>0),0]=0  #因为numpy.where(a[:,0]>0) 本身输出的就是行号，所以直接加逗号就OK了~