Question

我需要比较一堆具有不同尺寸的numpy数组，比如说：

a = np.array([1,2,3])
b = np.array([1,2,3],[4,5,6])
assert(a == b[0])

如果除了

之外我不知道a和b的形状，我怎么能这样做呢？

len(shape(a)) == len(shape(b)) - 1

我也不知道从b跳过哪个维度。我想使用np.index_exp，但这似乎对我没有帮助......

def compare_arrays(a,b,skip_row):
    u = np.index_exp[ ... ]
    assert(a[:] == b[u])

编辑或者换句话说，如果我知道阵列的形状和我想要错过的尺寸，我不想构造切片。如果我知道维度和位置的数量，放置“：”的位置以及放置“0”的位置，如何动态创建np.index_exp。

Answer 1

我只是在查看apply_along_axis和apply_over_axis的代码，研究它们如何构建索引对象。

让我们制作一个4d阵列：

In [355]: b=np.ones((2,3,4,3),int)

列出slices（使用列表*复制）

In [356]: ind=[slice(None)]*b.ndim

In [357]: b[ind].shape    # same as b[:,:,:,:]
Out[357]: (2, 3, 4, 3)

In [358]: ind[2]=2     # replace one slice with index

In [359]: b[ind].shape   # a slice, indexing on the third dim
Out[359]: (2, 3, 3)

或者使用您的示例

In [361]: b = np.array([1,2,3],[4,5,6])   # missing []
...
TypeError: data type not understood

In [362]: b = np.array([[1,2,3],[4,5,6]])

In [366]: ind=[slice(None)]*b.ndim    
In [367]: ind[0]=0
In [368]: a==b[ind]
Out[368]: array([ True,  True,  True], dtype=bool)

此索引与np.take基本相同，但同样的想法可以扩展到其他情况。

我不太关注:使用的问题。请注意，在构建索引列表时，我使用slice(None)。解释器将所有索引:转换为slice个对象：[start:stop:step] => slice(start, stop, step)。

通常您不需要使用a[:]==b[0]; a==b[0]就足够了。使用列表alist[:]复制，使用数组它什么也不做（除非在RHS上使用a[:]=...）。

未知维度的Numpy索引数组？

1 个答案: