Python Numpy Matrix應用元素行列輸入明智功能

Question

我想在NumPy矩陣的每個(i,j)條目上應用一個函數。 但我想使用值i和j 。 作為一個例子，給定向量A和B ，以及抽象函數f ，我想做以下事情：

for i in range(I):
    for j in range(J):
        M[i,j] = f(A[i],B[j])

而不是上面我可以做以下事情：

g = lambda (i,j): f(A[i],B[j])
apply(M, g)

只是試圖矢量化for循環。

謝謝！

Answer 1

這取決於功能

f = lambda x, y: x ** 2 + y

np.random.seed([3,1415])
A = np.random.rand(10)
B = np.random.rand(10)
M = np.empty_like(A)
M.fill(np.nan)

slc = np.array([3, 5, 6, 7])

然后做你提出的建議

M[slc] = f(A[slc], B[slc])

M

array([        nan,         nan,         nan,  1.14332569,         nan,
        0.11791531,  0.86916307,  0.86854943,         nan,         nan])

​