如何解决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
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。