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)