Требуется более быстрый код 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])