python : Numpy Matrix split

Question

我有一个形状为 4 x 129 的矩阵。我正在尝试进行水平拆分，如下所示:

In [18]: x = np.arange(4*129)

In [19]: x = x.reshape(4, 129)

In [20]: x.shape
Out[20]: (4, 129)

In [21]: y = np.hsplit(x, 13)

ValueError: array split does not result in an equal division

我知道它不能将它平均分成 13。我不想再用零填充一列并除以 13。

我想将 x 矩阵拆分为 13 个小矩阵，其中每 12 个拆分的大小应为 4 x 10，最后一个的大小应为4 x 9。

有什么办法可以做到这一点吗？

Answer 1

您可以为拆分传递索引，在这种情况下，您可以简单地使用 np.arange() 创建它们:

>>> a = np.hsplit(x, np.arange(12, 129, 12))
>>> 
>>> a[0].shape
(4, 12)
>>> a[-1].shape
(4, 9)