Skip to content

Instantly share code, notes, and snippets.

@pratapvardhan

pratapvardhan/benchmark_order_kth.py

Created July 23, 2014 16:55
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save pratapvardhan/f14f51ffcf757aea8bd9 to your computer and use it in GitHub Desktop.
Save pratapvardhan/f14f51ffcf757aea8bd9 to your computer and use it in GitHub Desktop.
Download ZIP
Benchmarking pandas order vs algos.kth_smallest
Raw
benchmark_order_kth.py
'''
Benchmarking pandas order vs algos.kth_smallest
'''
import pandas as pd
s = pd.Series(np.random.randint(100000,size=100000))
def fast_order(x, n):
v = pd.algos.kth_smallest(x.values.astype(float),n)
return x[x<=v]
s.order().head(10)
fast_order(s,10).order()
'''
%timeit s.order().head(10)
%timeit fast_order(s,10).order()
10 loops, best of 3: 61.4 ms per loop
100 loops, best of 3: 13.8 ms per loop
'''
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment