python - 零填充数组到下一个 2 的幂大小

Question

如Zero pad numpy array中所述，此函数对数组进行零填充：

A = np.array([1,2,3,4,5])
B = np.pad(A, (0, 3), 'constant')   # array([1, 2, 3, 4, 5, 0, 0, 0])

问题：是否有辅助函数numpy或scipy自动将数组补零到下一个 2 的幂大小？（用于 FFT 目的等）

当然我可以做类似的事情

def padpower2(A):
    return np.pad(A, (0, int(2**np.ceil(np.log2(len(A)))) - len(A)), 'constant')

但是记住/重写总是很棘手，所以有一个现成的numpy功能会更好。

标签： pythonarraysnumpy