Numpy数组切片和迭代

Question

我试图切片和迭代多维数组在同一时间。我有一个功能性的解决方案，但它有点丑，我打赌有一个我不知道的灵巧的方法来做迭代和切片。下面是代码:

import numpy as np
x = np.arange(64).reshape(4,4,4)
y = [x[i:i+2,j:j+2,k:k+2] for i in range(0,4,2) 
                          for j in range(0,4,2) 
                          for k in range(0,4,2)]
y = np.array(y)
z = np.array([np.min(u) for u in y]).reshape(y.shape[1:])

Answer 1

您的最后一次重塑不起作用，因为y没有定义形状。没有它，你得到:

>>> x = np.arange(64).reshape(4,4,4)
>>> y = [x[i:i+2,j:j+2,k:k+2] for i in range(0,4,2) 
...                           for j in range(0,4,2) 
...                           for k in range(0,4,2)]
>>> z = np.array([np.min(u) for u in y])
>>> z
array([ 0,  2,  8, 10, 32, 34, 40, 42])

但是尽管如此，你可能想要的是将数组重塑为6维，这将得到与上面相同的结果:

>>> xx = x.reshape(2, 2, 2, 2, 2, 2)
>>> zz = xx.min(axis=-1).min(axis=-2).min(axis=-3)
>>> zz
array([[[ 0,  2],
        [ 8, 10]],
       [[32, 34],
        [40, 42]]])
>>> zz.ravel()
array([ 0,  2,  8, 10, 32, 34, 40, 42])

Answer 2

在最后的意思中很难确切地告诉您想要什么，但是您可以使用stride_tricks来获得"更圆滑"的方法。这相当棘手。

import numpy.lib.stride_tricks
# This returns a view with custom strides, x2[i,j,k] matches y[4*i+2*j+k]
x2 = numpy.lib.stride_tricks(
        x, shape=(2,2,2,2,2,2),
        strides=(numpy.array([32,8,2,16,4,1])*x.dtype.itemsize))
z2 = z2.min(axis=-1).min(axis=-2).min(axis=-3)

不过，我不能说这是更好的可读性。(或者高效，因为每个min调用都会产生临时调用)

注意，我的答案与Jaime的不同，因为我试图匹配你的y元素。你可以告诉如果你用max替换min。