Question

请问您如何从xarray.DataArray中选择一列？这就是我做xarray的方式

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

my_data = np.random.rand(5,2)
da = xr.DataArray(my_data, 
                  coords={'my_id':np.arange(my_data.shape[0]),
                         'columns': ['x', 'y']},
                  dims=['my_id', 'columns'])

，我想选择column x。如果那是dataframe

df = pd.DataFrame(my_data, columns=['x','y'])

我正在寻找相当于df['x']

我定义DataArray的方式错误吗？

Answer 1

嗯，我想我明白了

da.sel(columns='x')

Answer 2

您可以使用docs中建议的类似大熊猫的.loc：

da.loc[:, ['x']]

<xarray.DataArray (my_id: 5, columns: 1)>
array([[ 0.534358],
       [ 0.113875],
       [ 0.905085],
       [ 0.96994 ],
       [ 0.548338]])
Coordinates:
  * my_id    (my_id) int32 0 1 2 3 4
  * columns  (columns) <U1 'x'

Answer 3

直接从文档中获取：http://xarray.pydata.org/en/stable/generated/xarray.DataArray.to_pandas.html

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

my_data = np.random.rand(5,2)
da = xr.DataArray(my_data, 
                  coords={'my_id':np.arange(my_data.shape[0]),
                         'columns': ['x', 'y']},
                  dims=['my_id', 'columns'])

df = da.to_pandas()

df['x']

结果

df ['x']

Out[27]: 
my_id
0    0.752056
1    0.048029
2    0.145835
3    0.083147
4    0.618351
Name: x, dtype: float64

从XArray中选择一列

3 个答案: