2015-09-23 16 views
1

我有以下4點矩陣的相應元素:numpy的:將標量函數不同的矩陣

w = [[w1, w2, w3, w4], [w5, w6, w7, w8], ...] 
x = [[x1, x2, x3, x4], [x5, x6, x7, x8], ...] 
y = [[y1, y2, y3, x4], [y5, y6, y7, y8], ...] 
z = [[z1, z2, z3, x4], [z5, z6, z7, z8], ...] 

這些是形狀(N,M)的。我想將一個標量函數f應用到每個矩陣的相應元素,使得得到的矩陣res也是n乘以m,並且res的第i,j個元素是結果func([[w[i,j], x[i,j]], [[y[i,j], z[i,j]])

很明顯,我可以嘗試遍歷每個元素,但是有沒有vectorized/numpy的實現呢?

+0

嗯。也許將'w','x','y'和'z'合併成一個'N×M×4'矩陣並應用'numpy.ix' ...現在太累了以便提供確切的細節。雖然它可能會很慢...需要考慮的事情。 – Dair

+0

對於完全一般的f(w,x,y,z),我認爲沒有辦法比for循環更有效。你能提供一些f的細節嗎? – AGML

回答