xarray中的乘法答案

【问题标题】：The multiplication in xarrayxarray中的乘法
【发布时间】：2021-05-26 12:08:15
【问题描述】：

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

【问题讨论】：

标签： python-3.x multiplication python-xarray

【解决方案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

【讨论】：