Appliquer la fonction de paire de coordonnées le long de l'axe notamment à l'aide de plusieurs variables dans Xarray

Question

My xarray Dataset a trois dimensions,

• x, y, t

et 2 variables,

• foo, bar

Je voudrais appliquer la fonction baz() sur chaque x, y coordonner les séries temporelles de paire t

baz() acceptera un tableau de foo -s et et tableau de bar -s pour une donnée (x, y)

J'ai du mal à comprendre si les structures intégrées pour gérer/distribuer cela existent dans xarray, pandas, numpy ou dask.

Des indices?

Mon approche actuelle est en train d'écrire un tableau python iterator, ou explorer ufunc:

• Les problèmes avec itérer sur quelque chose en python est que je dois traiter moi-même avec la concurrence et qu'il y a en tête dans l'itération . Le problème avec ufunc est que je ne le comprends pas assez et semble être une fonction qui est appliquée à des éléments individuels de la matrice, pas des sous-ensembles le long des axes.

La partie d'espoir de moi est à la recherche de quelque chose qui est xarray native ou daskable.

Answer 1

Sons (vaguement) comme vous cherchez Dataset.reduce:

ds.reduce(baz, dim=('x', 'y'))