0

我们知道 xarray 通过标签处理数据。即两个数据集相加或两个数据集相乘是在相同的维度和坐标下操作的。但是，我对 xarray 乘法运算后的结果形状有疑问。

代码如下所示：

import xarray as xr
import numpy as np
a = xr.DataArray([0,1,2,3],dims=['x'],
                 coords={'x':[10,20,30,40]})
b = xr.DataArray(np.array([[0,1,2,3],[0,100,200,300]]),
                 dims=['y','x'],
                 coords={'y':['y1','y2'],'x':[10,20,30,40]})
print(a*b)
print(b*a)

第一个结果是

<xarray.DataArray (x: 4, y: 2)>
array([[  0,   0],
       [  1, 100],
       [  4, 400],
       [  9, 900]])
Coordinates:
  * x        (x) int64 10 20 30 40
  * y        (y) <U2 'y1' 'y2'

第二个结果是

<xarray.DataArray (y: 2, x: 4)>
array([[  0,   1,   4,   9],
       [  0, 100, 400, 900]])
Coordinates:
  * y        (y) <U2 'y1' 'y2'
  * x        (x) int64 10 20 30 40

实际上，第一个和第二个是相等的，因为 xarray 是基于标签的，但是，为什么第一个结果的形状是 [4,2] 而第二个是 [2,4]？谁能告诉我？谢谢！

4

1 回答 1

2

(a*b.T)并且(b.T*a)都产生具有形状的数组[4,2]

我相信这是因为b.T改变了 DataArray 维度顺序b而不是坐标顺序。

比较 b 和 bT：

print(b)

<xarray.DataArray (y: 2, x: 4)>
array([[  0,   1,   2,   3],
       [  0, 100, 200, 300]])
Coordinates:
  * y        (y) <U2 'y1' 'y2'
  * x        (x) int64 10 20 30 40

print(b.T)

<xarray.DataArray (x: 4, y: 2)>
array([[  0,   0],
       [  1, 100],
       [  2, 200],
       [  3, 300]])
Coordinates:
  * y        (y) <U2 'y1' 'y2'
  * x        (x) int64 10 20 30 40
于 2021-03-23T17:51:34.930 回答