如何获取 xr.Dataset 坐标作为字符串？

Question

我创建了一个 xarray.Dataset。在Dataset中，我想获取某个corrdinate的列表。

示例代码如下所示。代码中，数据集'ds'包含'MachineType'中的4个，即'MC_A'、'MC_B'、'MC_C'、'MC_D'。我想将它们作为字符串获取，以便最后一个 for 循环执行 'MC_A'、'MC_B'、'MC_C'、'MC_D'.

import xarray as xr
import numpy as np
import pandas as pd

x = np.linspace(0,1500,16)
y1 = np.random.random((1,1,1,1,1,1,6,len(x)))
y2 = np.random.random((1,1,1,1,1,1,6,len(x)))
y3 = np.random.random((1,1,1,1,1,1,6,len(x)))
y4 = np.random.random((1,1,1,1,1,1,6,len(x)))

da1 = xr.DataArray(y1, dims=('MachineID','MachineType','Year','Month','Compensation','Axis','SensorID','Pos'),
                       coords={'MachineID':['MC_A'],'MachineType':['Car'],'Year':['2020'],'Month':['Aug'],'Compensation':[1],
                            'Axis':['X'],'SensorID':range(6),'Pos':x},
                       name='val')

da2 = xr.DataArray(y2, dims=('MachineID','MachineType','Year','Month','Compensation','Axis','SensorID','Pos'),
                       coords={'MachineID':['MC_B'],'MachineType':['Car'],'Year':['2020'],'Month':['Aug'],'Compensation':[1],
                            'Axis':['X'],'SensorID':range(6),'Pos':x},
                       name='val')

da3 = xr.DataArray(y3, dims=('MachineID','MachineType','Year','Month','Compensation','Axis','SensorID','Pos'),
                       coords={'MachineID':['MC_C'],'MachineType':['Train'],'Year':['2020'],'Month':['Aug'],'Compensation':[1],
                            'Axis':['X'],'SensorID':range(6),'Pos':x},
                       name='val')

da4 = xr.DataArray(y4, dims=('MachineID','MachineType','Year','Month','Compensation','Axis','SensorID','Pos'),
                       coords={'MachineID':['MC_D'],'MachineType':['Train'],'Year':['2020'],'Month':['Aug'],'Compensation':[1],
                            'Axis':['X'],'SensorID':range(6),'Pos':x},
                       name='val')

#%%
df = pd.concat([da1.to_dataframe(), da2.to_dataframe(), da3.to_dataframe(), da4.to_dataframe()])
ds=xr.Dataset.from_dataframe(df)
ds


#%% This part is not working as I want
for machineid in ds['MachineID']:
    print(machineid)

Answer 1

只需在打印命令中添加.values并将类型更改为str:

for machineid in ds['MachineID']:
    print(str(machineid.values))

如何获取 xr.Dataset 坐标作为字符串？

How to get the xr.Dataset coordinates as string?

python

dataset

python-xarray