在无序对的函数上向量化一个总和?

时间:2015-10-15 18:27:47

标签: python numpy vectorization

我希望将无序对(n,m)上的函数f与n和m distinct相加,其中n,m的范围从-N到N.

该函数依赖于存储在字典中的浮点序列,从-N到N索引。它看起来像:

def f(n,m):
    return dict[n]*dict[m]*np.sin(c*(n-m)) / (n-m)

天真地,我写道:

indices = [ (n,m) for n in range(-N,N+1) for m in range(-N,N+1) if m>n ]
sum([f(n,m) for (n,m) in indices])

有没有办法用numpy对它进行矢量化并使其更有效率?我使用Python 3,以防万一。

0 个答案:

没有答案
相关问题