2

我正在尝试使用我的 xarray 坐标进行坐标转换,所以

我有一个像这样的 DataArray:

d = xr.DataArray(np.zeros((10, 10, 1)), dims=['x', 'y', 'z']

并且正在执行以下操作:

r = np.sqrt(d.x**2 + d.y**2 + d.z**2)
theta = np.arctan2(np.sqrt(d.x**2 + d.y**2), d.z)
phi = np.arctan2(d.y, d.x)

我得到以下形状:

In [212]: r.shape
Out[212]: (10, 10, 1)

In [214]: theta.shape
Out[214]: (10, 10)

In [216]: phi.shape
Out[216]: (10,)

我想看看(10, 10, 1)我为所有人得到的 r 形状。

看起来 xarray 的花哨广播只适用于基本的算术运算,而 numpy 函数正在天真地广播。

它是否正确?

有没有经过批准的方法来解决这个问题?我可以编写我自己的 xarray 感知 arctan2 函数来获得正确的广播,但我希望我不必这样做。

4

1 回答 1

2

不幸的是,NumPy 函数通常不会对 xarray 对象进行适当的广播。但是 xarray 确实带有许多这些函数的包装版本,包括arctan2xarray.ufuncsmodule中。所以xarray.ufuncs.arctan2应该做你正在寻找的东西。

于 2016-10-05T21:49:52.397 回答