Skip to content

Instantly share code, notes, and snippets.

@mitliagkas

mitliagkas/truncate.py

Last active August 29, 2015 14:01
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 mitliagkas/a9088a183a68e94ee497 to your computer and use it in GitHub Desktop.
Save mitliagkas/a9088a183a68e94ee497 to your computer and use it in GitHub Desktop.
Download ZIP
Truncate vector x, only keeping the s elements of largest magnitude.
Raw
truncate.py
part = np.argpartition(abs(x),len(x)-s,axis=0)
x[part[:-s]] = 0
@mitliagkas
Copy link
Author

Now faster!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment