需要更快的 python 代码来计算样本熵

Question

这是我在为样本熵编写python代码时遇到的问题。

map(max, abs(a[i]-a) ) 很慢。

还有其他功能比map() 性能更好吗？

a 是 ndarray，看起来像 np.array([ [1,2,3,4,5],[2,3,4,5,6],[3,4,5,3,2] ])

Answer 1

使用矢量化最大值

>>> map(max, abs(a[2]-a) )
[3, 4, 0]
>>> np.abs(a[2] - a).max(axis=1)
array([3, 4, 0])