numpy.max or max ? Which one is faster?

2020-06-06 17:41发布

问题:

In python, which one is faster ?

numpy.max(), numpy.min()

or

max(), min()

My list/array length varies from 2 to 600. Which one should I use to save some run time ?

回答1:

Well from my timings it follows if you already have numpy array a you should use a.max (the source tells it's the same as np.max if a.max available). But if you have built-in list then most of the time takes converting it into np.ndarray => that's why max is better in your timings.

In essense: if np.ndarray then a.max, if list and no need for all the machinery of np.ndarray then standard max.



回答2:

I was also interested in this and tested the three variants with perfplot (a little project of mine). Result: You're not going wrong with a.max().

Code to reproduce the plot:

import numpy
import perfplot

perfplot.show(
    setup=lambda n: numpy.random.rand(n),
    kernels=[max, numpy.max, lambda a: a.max()],
    labels=["max(a)", "numpy.max(a)", "a.max()"],
    n_range=[2 ** k for k in range(25)],
    logx=True,
    logy=True,
    xlabel="len(a)",
)


回答3:

It's probably best if you use something like the Python timeit module to test it for yourself. That way you can test your own data in your own environment, rather than relying on third parties with various test data and environments which aren't necessarily representative of yours.



回答4:

numpy.min and numpy.max have slightly different semantics (and call signatures) to the builtins, so the choice shouldn't be to do with speed. Use the numpy versions if you need to be able to handle multidimensional data sanely. If you're just using Python lists or other things that don't know about dimensionality, use the builtins.