def test_array_sort_indices():
arr = pa.array([1, 2, None, 0])
result = pc.array_sort_indices(arr)
assert result.to_pylist() == [3, 0, 1, 2]
result = pc.array_sort_indices(arr, order="ascending")
assert result.to_pylist() == [3, 0, 1, 2]
result = pc.array_sort_indices(arr, order="descending")
assert result.to_pylist() == [1, 0, 3, 2]
with pytest.raises(ValueError, match="not a valid order"):
pc.array_sort_indices(arr, order="nonscending")
def argsort(
self,
ascending: bool = True,
kind: str = "quicksort",
na_position: str = "last",
*args,
**kwargs,
) -> np.ndarray:
order = "ascending" if ascending else "descending"
null_placement = {
"last": "at_end",
"first": "at_start"
}.get(na_position, None)
if null_placement is None or pa_version_under7p0:
# Although pc.array_sort_indices exists in version 6
# there's a bug that affects the pa.ChunkedArray backing
# https://issues.apache.org/jira/browse/ARROW-12042
fallback_performancewarning("7")
return super().argsort(ascending=ascending,
kind=kind,
na_position=na_position)
result = pc.array_sort_indices(self._data,
order=order,
null_placement=null_placement)
if pa_version_under2p0:
np_result = result.to_pandas().values
else:
np_result = result.to_numpy()
return np_result.astype(np.intp, copy=False)
