子分组numpy阵列

子分组numpy阵列,numpy,Numpy,我想将ad=(d1,…,dn)形状的numpy数组a子集,只得到r=(r1,…,rn)第一主辅,即所有值a[i1,…,in],这样ij

我想将a
d=(d1,…,dn)
形状的numpy数组
a
子集,只得到
r=(r1,…,rn)
第一主辅,即所有值
a[i1,…,in]
,这样
ij
。当然,生成的数组具有shape
r

我该怎么做?我尝试了
A[:r]
,但当然没有成功:)。正确的方法应该是
A[:r1,…:rn]
,但是
len(A.shape)
是未知的,所以我不能逐个编写所有子集

你知道我如何得到这个子矩阵吗?

你可以使用

A[tuple([slice(None,i)表示r中的i])]
您可以使用

A[tuple([slice(None,i)表示r中的i])]

Perfec,谢谢。不知道slice()函数!内部的一对方括号是不必要的,但在其他方面看起来不错。谢谢。不知道slice()函数!内部的一对方括号是不必要的,但在其他方面看起来不错。