0

我接到了将一段 MATLAB 代码翻译成 IDL 的任务，当我遇到 MATLAB 函数 accumarry() 时遇到了障碍。此处描述的函数 用于根据另一个数组中给出的索引对一个数组中的元素求和。示例 1 可能比页面顶部的实际功能描述更好地解释了这一点。在尝试在 IDL 中重现示例 1 时，我无法避免 for 循环，但我相信这是可能的。我最好的尝试如下：

vals = [101,102,103,104,105]
subs = [0,1,3,1,3]

n = max(subs)+1
accum = make_array(n)

for i = 0, n-1 do begin
   wVals = where(subs eq i,count)
   accum[i] = count eq 0 ? 0 : total(vals[wVals])
endfor

print,accum
;       101.000      206.000      0.00000      208.000

任何有关改进这一点的建议将不胜感激！我希望 IDL 有一个类似的内置功能，但一直没能找到。也许直方图分箱有一些魔力？

4

1 回答 1

0

我在 Coyote 的 IDL 网站上找到了许多可能的解决方案（不足为奇。） http://www.idlcoyote.com/code_tips/drizzling.html

我最终使用了以下内容，作为性能和可读性之间的折衷：

function accumarray,data,subs

mx = max(subs)
accum = fltarr(mx+1) 

h = histogram(subs,reverse_indices=ri,OMIN=om)
for j=0L,n_elements(h)-1 do if ri[j+1] gt ri[j] then $
   accum[j+om] = total(vals[ri[ri[j]:ri[j+1]-1]])

return,accum
于 2013-06-03T11:03:07.717 回答