python-3.x - xarray 中的乘法

Question

我们知道 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]？ 谁能告诉我吗？谢谢!

Answer 1

(a*b.T) 和 (b.T*a) 都生成形状为 [4,2]

的数组

我相信这是因为 b.T 改变了 b 中的 DataArray 维度顺序，但没有改变坐标顺序。

将 b 与 b.T 进行比较:

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