从数组中提取矩阵数组的最合适方法

Question

numpy的新用户。我有一个问题，我没有找到一个不涉及原始循环的解决方案。

我有一个大小为N * M的一维数组。每个大小为M的块都由一个平面矩阵和一些附加值组成。我想提取th​​oses矩阵。

对于该示例，让我们考虑一下该表单的一维数组：

[a1 b1 c1 d1 e1 f1 g1 a2 b2 c2 d2 e2 f2 g2 .... aN bN cN dN eN fN gN]

构建以下N 2 * 2矩阵数组的最优雅和/或最有效的方法是什么：

[[[a1,b1],
  [c1,d1]],
 [[a2,b2],
  [c2,d2]],
 ....
 [[aN,bN],
  [cN,dN]]]

请注意，e1 f1 g1已被丢弃。

作为一个具体的例子，考虑：

import numpy as np
x = np.array([0,1,2,3,-1,-1,-1
              0,1,2,3,-1,-1,-1])

预期结果应为包含两个矩阵[[0,1]，[2,3]]的数组

Answer 1

重新整形后跟切片的简单示例：

In [219]: x = np.arange(6)                                                      
In [220]: np.reshape(x,(2,3))                                                   
Out[220]: 
array([[0, 1, 2],
       [3, 4, 5]])
In [221]: np.reshape(x,(2,3))[:,:2]                                             
Out[221]: 
array([[0, 1],
       [3, 4]])

Answer 2

一种解决方案似乎使用as_strided函数。

让b为块大小，以下工作原理：

y = np.lib.stride_tricks.as_strided(x, shape=(x.size // b,2,2),
                                    strides=(x.strides[0]*b,
                                             x.strides[0]*2,
                                             x.strides[0]))

例如：

>>> import numpy as np
>>> x = np.array([0,1,2,3,-1,-1,-1,
...               0,1,2,3,-1,-1,-1,
...               0,1,2,3,-1,-1,-1])
>>> b = 7
>>> y = np.lib.stride_tricks.as_strided(x, shape=(x.size // b,2,2),
...                                         strides=(x.strides[0]*b,
...                                                  x.strides[0]*2,
...                                                  x.strides[0]))
>>> print(y)
[[[0 1]
  [2 3]]

 [[0 1]
  [2 3]]

 [[0 1]
  [2 3]]]

如果有人有更好的解决方案，请告诉我;感谢hpaulj花时间帮忙。