5

This is the problem I have faced when I am writing python code for sample entropy.

map(max, abs(a[i]-a) ) is very slow.

Is there any other function perform better than map() ?

Where a is ndarray that looks like np.array([ [1,2,3,4,5],[2,3,4,5,6],[3,4,5,3,2] ])

Daniel
  • 19,179
  • 7
  • 60
  • 74
Xin Li
  • 61
  • 5

1 Answers1

6

Use the vectorized max

>>> map(max, abs(a[2]-a) )
[3, 4, 0]
>>> np.abs(a[2] - a).max(axis=1)
array([3, 4, 0])
Viktor Kerkez
  • 45,070
  • 12
  • 104
  • 85