xarray 中的乘法

问题描述

我们知道 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,[0,100,200,300]]),dims=['y','x'],coords={'y':['y1','y2'],'x':[10,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,4,9],[  0,400,900]])
Coordinates:
  * y        (y) <U2 'y1' 'y2'
  * x        (x) int64 10 20 30 40

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

解决方法

(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