好的方法来“扩展”一个numpy ndarray?

7 浏览
0 Comments

好的方法来“扩展”一个numpy ndarray?

有没有好的方法可以“扩展”一个numpy的ndarray?比如说我有一个像这样的ndarray:

[[1 2]
 [3 4]]

我希望每一行都包含更多的元素,通过填充零:

[[1 2 0 0 0]
 [3 4 0 0 0]]

我知道肯定有一些蛮力的方法可以做到这一点(比如构建一个更大的带有零的数组,然后从旧的较小的数组中复制元素),只是想知道有没有更Pythonic的方法。尝试过numpy.reshape但是没有成功:

import numpy as np
a = np.array([[1, 2], [3, 4]])
np.reshape(a, (2, 5))

Numpy报错:ValueError: total size of new array must be unchanged

0