Python增加特定的矩陣元素

Question

我想識別矩陣中的某些元素（也就是說小於4）並將元素增加1000。一個元素是2，所以這個元素更改爲1002.我怎樣纔能有效地做到這一點？

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] 

最終x應np.array([[1002,1003,1004,5],[1003,1004,5,6],[1004,5,6,7],[8,9,10,11]])

Answer 1

你可以做這樣的事情：

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]]