REF/DOC: use take (a bit faster), timing in docstring