對函數使用多維數組參數的索引

Question

我遇到以下問題並且找不到任何解決方案。

for index, value in numpy.ndenumerate(re): 
     re[index] += rec[(index)] 

這不起作用。功能如下所示：

def rec(x1, x2, x3, y1, y2, y3, z1, z2, z3): 

索引的大小確實適合該函數的參數數量。我想盡可能保持它一樣，因此

for [x1][x2][x3][y1][y2][y3][z1][z2][z3], value in numpy.ndenumerate(re): 
     re[index] += rec[(index)] 

不幫助我。你有什麼想法嗎？

Answer 1

This doesn't work unfortunately是不是描述你的問題的正確方法。你需要告訴我們一些你得到的錯誤。

但我的猜測與rec有關，不是可轉位的，或者有__getitem__方法。

你定義一個函數有很多參數：

def rec(x1, x2, x3, y1, y2, y3, z1, z2, z3): 
    .... 

你就必須調用它的功能。它不是一個列表或數組。

rec(1,2,3,....) # should work 

或

index = (1,2,3,...) 
rec(*index) 

如果是這樣，那麼下面可能的工作：

for index, value in numpy.ndenumerate(re): 
     re[index] += rec(*index) 

Answer 2

二維例子（IPython的筆記本）

import numpy as np 
import matplotlib.pyplot as p # just for visualization 
%matplotlib inline 

x=np.arange(0,1,0.01) # <-- give ranges to your parameters 
y=np.arange(0,1,0.01) 
xv, yv = np.meshgrid(x,y) # <--- generate the n-dimensional indeces 

zv= np.sin(xv + 3* yv) # <-- this is the ufunc, look ma: no for loops 

p.imshow(zv,aspect ='1')