`.sel` with a bare number silently selects by a different scale on a coordinate in degrees or percent

Aberta
  • Algo errado?
#400 0 comentários 0 reações 0 responsáveis Ver no GitHub

Ninguém assumiu esta issue ainda.

Avaliação

Dificuldade
4/5
Tempo estimado
3-5 dias
Facilidade para iniciantes
56/100
Tipo de issue
Bug
Clareza
Claramente especificada
Status de atividade
Ativa
Stack de tecnologia
python
Domínio
database

Direção de pesquisa

Start by reading PintIndex.sel and conversion.convert_indexer_units, then trace how array_convert_units handles bare numbers for index coordinates. Compare selection on scaled units such as degree and percent with radian and dimensionless, and review the existing indexer conversion tests. Done means the behavior is explicitly resolved and covered by tests for nearest selection and slices.

Escrita pelo modelo de indexação a partir do texto da issue.

Descrição

Hello. Thanks for all your work on pint-xarray, it's awesome.

To be up front, this issue was identified using AI (Claude Opus 5.5) - I was only really looking at quantified data, not coords, but this came up during testing. This doesn't affect me directly but seems like a case for a warning & a docs update (which I'm happy to do if you agree).

Everything below was written & tested using AI. Cheers!

What happened

I understand that a bare number on a quantified index counts as dimensionless, and that a Quantity is the intended label.

The problem is that on a coordinate whose unit is compatible with dimensionless but scaled, such as degree, that rule gives no error or warning, and code that worked before .pint.quantify() silently selects different labels.

import pint_xarray  # noqa: F401
import xarray as xr

da = xr.DataArray(
    [1.0, 2.0, 3.0],
    dims="lat",
    coords={"lat": ("lat", [10.0, 20.0, 30.0], {"units": "degree"})},
)
quantified = da.pint.quantify()

print(da.sel(lat=20.0, method="nearest").lat.item())  # 20.0
print(quantified.sel(lat=20.0, method="nearest").lat.item())  # 30.0

print(da.sel(lat=slice(10.0, 20.0)).sizes["lat"])  # 2
print(quantified.sel(lat=slice(10.0, 20.0)).sizes["lat"])  # 0

What I expected

An error, as on a coordinate in metres, where the same mistake is caught:

xr.DataArray(
    [1.0, 2.0],
    dims="z",
    coords={"z": ("z", [10.0, 20.0], {"units": "m"})},
).pint.quantify().sel(z=20.0)
# PintExceptionGroup: Cannot convert from 'dimensionless' (dimensionless) to 'meter' ([length])

Cause

PintIndex.sel passes its labels to conversion.convert_indexer_units, which ends in array_convert_units.
That treats a value that is not a quantity as dimensionless, as its docstring says and test_convert_indexer_units expects, and pint converts dimensionless to degrees by treating the number as radians.

So 20.0 becomes about 1146 degrees before the lookup:

  • method="nearest" then finds the largest label, 30.
  • The slice becomes roughly 573 to 1146 degrees, which is empty.
  • An exact .sel(lat=20.0) raises KeyError, so that case is at least loud.

.sel with a real quantity is correct, as in quantified.sel(lat=20.0 * ureg.degree, method="nearest").

Which units are affected

Run on the same three labels, 10, 20 and 30:

Coordinate unit .sel(c=20.0, method="nearest") .sel(c=slice(10.0, 20.0)) size
degree 30 0
arcminute 30 0
percent 30 0
radian 20 2
dimensionless 20 2
Unquantified, for comparison 20 2

The accessor gives a different answer from the index: quantified.pint.sel(lat=20.0, method="nearest") returns the 20 degree row, since it has no indexer units to convert to and strips the coordinate's own.

Latitude and longitude are the common case, since degrees_north is compatible with dimensionless under cf-xarray's registry, and .sel(lat=..., method="nearest") is routine.

Question

Is this the behaviour you intend?
If it is, could the docs say so, since docs/creation.rst still says xarray does not support units in indexes, and nothing I could find says that labels need to be quantities?

If not, one option is to raise for a bare number on an index with a scaled unit, matching what already happens with metres.
Selection on dimensionless and radian could keep working, since the number means the same either way.

Versions

Reproduced in a clean environment with only pint-xarray and xarray installed:

Package Version
Python 3.12
pint-xarray 0.6.1
pint 0.26.2
xarray 2026.9.0
numpy 2.5.4

convert_indexer_units and array_convert_units are unchanged on main.

The closest existing issues I found are #303, about method= being rejected, and #276, about indexing with quantities, which both concern a different failure.

Linguagem predominante
Python
Estrelas
120
Forks
17
Nenhum PR com merge em 30d

Preparar o ambiente

Primeiros passos

  1. Leia a issue inteira e depois o guia de contribuição do projeto.
  2. Comente na issue dizendo que vai assumir — evita que duas pessoas façam o mesmo trabalho.
  3. Faça um fork do repositório e trabalhe em uma branch.
  4. Abra um pull request que referencie o número da issue.

Mais de xarray-contrib/pint-xarray

Todas as issues de xarray-contrib/pint-xarray

Issues semelhantes

Mais issues de Python

Receba novas issues na sua caixa de entrada

Um resumo curto de issues do GitHub para quem está começando.