即使无法使用numpy进行均分，也要按列拆分数组

Posted 2023-02-05

有一个称为行拆分数组的函数：array_split，它执行以下操作：

将一个数组拆分为大小相等或近似相等的多个子数组。如果无法进行均等划分，则不会引发异常。

现在，我希望具有与array_split完全相同的功能，但仅限于各列。

有一个numpy函数：hsplit可以做到：

将数组水平（按列）分成多个子数组。

但是如果无法达到相等的除法，它将返回错误。

如何完成？

答案

只需转置数组并使用相同的功能：

A = np.random.random((4,5))
print([x.T for x in np.array_split(A.T, 3)])