DataArray elimina atributos en cálculos simples

Answer

1. 

   davy.ai

   July 16, 2023 at 6:33 am

   Al realizar cálculos simples en un DataArray de xArray, como multiplicación o suma, se crea un nuevo DataArray con los valores modificados, pero sin los atributos originales. Esto se debe a que los atributos no se consideran parte de la operación matemática, sino metadatos asociados con el array original.

   Sin embargo, al utilizar funciones específicas de numpy como .round(x), se conservan los metadatos originales porque el método se aplica directamente al DataArray original en lugar de crear uno nuevo.

   Para multiplicar un DataArray sin perder sus atributos, una opción es aplicar la operación de multiplicación como una función de numpy utilizando el método .apply_ufunc. Este método le permite especificar los tipos de datos de entrada y salida, así como cualquier metadato adicional, como atributos, que se retendrá en el DataArray de salida.

   Por ejemplo:

   import numpy as np
   import xarray as xr
   
   ejemplo = xr.DataArray(np.array([1,2,3]), attrs={'uno':1})
   multiplicacion = np.multiply(ejemplo, 3)
   con_atributos = xr.DataArray(multiplicacion, attrs=ejemplo.attrs)
   

   En este ejemplo, se utiliza la función multiply de numpy para realizar la multiplicación y el array resultante se convierte nuevamente en un DataArray de xArray con los atributos originales agregados.

Comments are closed.