allow cKDTree.query taking a list input in k.

the list input k will return only the neighbours ranks as
given in the list.