[英]Extracting 2 D from 3D numpy array
我有两个格式为(1000、1000、20)的3D数组。 最后一个维度13是时间戳的索引。 我想通过时间戳逐步遍历数组并比较数组。 假设我有A(1000,1000,20)和B(1000,1000,20)。
我想要类似的东西
for t in range(0,21):
asub = A[,,t]
bsub = B[,,t]
#compare asub and bsub
但是,该语法似乎不起作用。 我怎样才能做到这一点?
documentation
,您需要“合并高级索引和基本索引” 。 因此, 高级 indexing
涉及对array
特定axis
进行indexing
。
例如,从index
1
开始获取elements
:
>>> a = np.array([[1,2,3], [4,5,6],[7,8,9]])
>>> a[1:, 1:]
array([[5, 6],
[8, 9]])
所以,如果你想获得的element
从第三axis
在index
t
,你需要选择的所有elements
与其它axis
与只是一个普通的冒号( :
),然后指定t
最后index
:
所以你想做:
A[:, :, t]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.