2014-09-02 122 views
0

我想识别矩阵中的某些元素(也就是说小于4)并将元素增加1000。一个元素是2,所以这个元素更改为1002.我怎样才能有效地做到这一点?Python增加特定的矩阵元素

import numpy as np 

x = np.array([[2,3,4,5],[3,4,5,6],[4,5,6,7],[8,9,10,11]]) 
x[x <= 4] = x[x = x+1000] 

最终xnp.array([[1002,1003,1004,5],[1003,1004,5,6],[1004,5,6,7],[8,9,10,11]])

回答

4

你可以做这样的事情:

import numpy as np 
x = np.array([[2,3,4,5],[3,4,5,6],[4,5,6,7],[8,9,10,11]]) 
x[x <= 4] += 1000 
print x 

输出:

[[1002 1003 1004 5] 
[1003 1004 5 6] 
[1004 5 6 7] 
[ 8 9 10 11]]