python numpy将数组拆分为不相等的子数组


问题内容

我试图将数组拆分为n个部分。有时这些部分的大小相同,有时它们的大小不同。

我正在尝试使用:

split = np.split(list, size)

当大小均分为列表时,此方法工作正常,否则失败。有没有一种方法可以用额外的“很少”元素来“填充”最终的数组?


问题答案:
def split_padded(a,n):
    padding = (-len(a))%n
    return np.split(np.concatenate((a,np.zeros(padding))),n)