Início / coding / Perguntas / 78054191
Accepted
ClimateUnboxed
Asked: 2024-02-25 06:17:43 +0800 CST

como usar uma variável de string para definir o nome do dataarray ao adicionar ao conjunto de dados xarray? [duplicado]

  • 772

Quero adicionar a dataarraya an xarray datasete fazê-lo usando xarray.assign, mas não sei como definir o nome do dataarray usando uma variável de string (ou seja, para chamar a nova entrada "myvar":

import xarray as xr

varname="myvar"
vals=[1,2,3]
coords=[4,5,6]
ds=xr.Dataset(data_vars={},coords={'xcoord':coords})
ds=ds.assign(varname=(['xcoord'],vals))
ds.to_netcdf("test.nc")

ds.close()

Isso me dá a variável chamada "varname" - como faço para usar uma variável aqui?

python

1 respostas

  1. Best Answer

    Construa um dicionário onde varnameestá a chave:

    kwargs = {varname: (['xcoord'],vals)}

    E então passe esse dicionário como argumento para ds.assign(), usando **a sintaxe:

    ds=ds.assign(**kwargs)

    Se quiser, você pode fazer tudo de uma só vez, sem criar um dicionário intermediário:

    ds=ds.assign(**{varname: (['xcoord'],vals)})
    • 1

relate perguntas