allow cKDTree.query taking a list input in k.

the list input k will return only the neighbours ranks as
given in the list.
This says you changed the return shape for ``k = 1``; the last dimension of size 1 was squeezed. Is that right? If so, that's a backwards compatibility issue.