boolean-indexing code snippets in python