如何識別xarray中的時間，長度和緯度坐標？

Question

確定xarray dataArray對象的哪些坐標包含longitude ， latitude和time的最佳方法是什么？

典型的dataArray可能如下所示：

<xarray.Dataset>
Dimensions:    (ensemble: 9, lat: 224, lon: 464, time: 12054)
Coordinates:
  * lat        (lat) float64 25.06 25.19 25.31 25.44 ... 52.56 52.69 52.81 52.94
  * lon        (lon) float64 -124.9 -124.8 -124.7 ... -67.31 -67.19 -67.06
  * time       (time) datetime64[ns] 1980-01-01 1980-01-02 ... 2012-12-31
Dimensions without coordinates: ensemble
Data variables:
    elevation  (lat, lon) float64 dask.array<shape=(224, 464), chunksize=(224, 464)>
    temp       (ensemble, time, lat, lon) float64 dask.array<shape=(9, 12054, 224, 464), chunksize=(1, 287, 224, 464)>

一種方法可能是循環變量coords標識的變量，如temp.coords ，尋找time ， longitude和latitude的standard_name屬性。 但是許多數據集似乎並不包含所有變量的standard_name屬性。

我猜測另一種方法是尋找在units屬性，並試圖找出他們是否有合適的units屬性（例如degrees_east或degrees_west為longitude ，等）。

有沒有更好的辦法？

Answer 1

MetPy包中包含一些用於系統坐標識別的幫助器。 您可以在MetPy教程的xarray中看到它的工作原理。 例如，如果您想要一個名為temp的DataArray的時間坐標（假設它來自已被MetPy解析的數據集），您只需調用：

temp.metpy.time

這是通過根據CF約定解析坐標元數據在內部完成的。

這是一個簡短的例子：

import xarray as xr
import metpy.calc as mpcalc

ds = xr.tutorial.load_dataset('air_temperature')
ds = ds.metpy.parse_cf()

x,y,t = ds['air'].metpy.coordinates('x','y','time')

print([coord.name for coord in (x, y, t)])

產生：

['lon', 'lat', 'time']

Answer 2

您可以使用xarray filter_by執行類似下面的代碼：

def x_axis(nc):
    xnames = ['longitude', 'grid_longitude', 'projection_x_coordinate']
    xunits = [
        'degrees_east',
        'degree_east',
        'degree_E',
        'degrees_E',
        'degreeE',
        'degreesE',
    ]
    xvars = list(set(
        nc.get_variables_by_attributes(
            axis=lambda x: x and str(x).lower() == 'x'
        ) +
        nc.get_variables_by_attributes(
            standard_name=lambda x: x and str(x).lower() in xnames
        ) +
        nc.get_variables_by_attributes(
            units=lambda x: x and str(x).lower() in xunits
        )
    ))
    return xvars

Answer 3

如果您只是尋找充當索引的特殊坐標，那么您可以迭代ds.indexes並對其名稱進行一些字符串解析。 就像是：

ds = xr.tutorial.load_dataset('air_temperature')
ds.lat.attrs.pop('standard_name')

for k in ds.indexes.keys():
    v = ds[k]
    sn = v.attrs.get('standard_name')
    if not sn:
        if 'lon' in k:
            v.attrs.update(standard_name='longitude')
            continue
        if 'lat' in k:
            v.attrs.update(standard_name='latitude')
            continue
        if 'time' in k or k in ['day', 't', 'month', 'year']:
            v.attrs.update(standard_name='time')

Answer 4

我認為我們應該嚴重依賴CF慣例。 它們正是出於這個原因而存在。 所以我建議將這個問題分成兩部分：

• 修復非CF投訴數據集（可能是一個用於此目的的小型庫是有意義的 - 它可能包含將公共變量名稱轉換為適當的standard_name屬性的邏輯）
• 解析CF投訴數據集（可以利用standard_name屬性）