mitliagkas/truncate.py

Last active August 29, 2015 14:01

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/mitliagkas/a9088a183a68e94ee497.js"></script>
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

Author

mitliagkas commented Feb 12, 2015

Now faster!

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