需要更快的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])