Mathematica快速2D分箱算法

Mathematica快速2D分箱算法,第1张

Mathematica快速2D分箱算法

这是一种基于Szabolcs帖子的方法,速度大约快一个数量级

data = RandomReal[5, {500000, 3}];(*500k values*)zvalues = data[[All, 3]];epsilon = 1*^-10;(*prevent 101 index*)(*rescale and round (x,y) coordinates to index pairs in the 1..100 range*)indexes = 1 + Floor[(1 - epsilon) 100 Rescale[data[[All, {1, 2}]]]];res2 = Module[{gb = GatherBy[Transpose[{indexes, zvalues}], First]},     SparseArray[     gb[[All, 1, 1]] ->       Total[gb[[All, All, 2]], {2}]]]; // AbsoluteTiming

给出{2.012217,Null}

AbsoluteTiming[ System`SetSystemOptions[   "SparseArrayOptions" -> {"TreatRepeatedEntries" -> 1}]; res3 = SparseArray[indexes -> zvalues]; System`SetSystemOptions[   "SparseArrayOptions" -> {"TreatRepeatedEntries" -> 0}]; ]

给出{0.195228,空}

res3 == res2True

“ TreatRepeatedEntries”-> 1将重复的位置加起来。



欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5175545.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-11-18
下一篇 2022-11-18

发表评论

登录后才能评论

评论列表(0条)

保存