numpy的新用户。我有一个问题,没有找到不涉及原始循环的解决方案。
我有一个大小为N * M的一维数组。大小为M的每个块都由一个平面矩阵和一些其他值组成。我想提取那些矩阵。
对于此示例,让我们考虑以下形式的一维数组:
None
构建以下N 2 * 2矩阵数组的最优雅和/或最有效的方法是什么:
[a1 b1 c1 d1 e1 f1 g1 a2 b2 c2 d2 e2 f2 g2 .... aN bN cN dN eN fN gN]
请注意,e1 f1 g1已被丢弃。
作为一个具体示例,请考虑:
[[[a1,b1],
[c1,d1]],
[[a2,b2],
[c2,d2]],
....
[[aN,bN],
[cN,dN]]]
预期结果应为包含两个矩阵[[0,1],[2,3]]
的数组答案 0 :(得分: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]])
答案 1 :(得分:0)
一种解决方案似乎使用了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抽出宝贵的时间来提供帮助。