numpy
code
>>> a = np.arange(6).reshape(2,3)
>>> np.pad(a, ((0,2), (1,1)), "constant")
array([[0, 0, 1, 2, 0],
[0, 3, 4, 5, 0],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0]])
解释
- pad第二个参数,原来的数组有几个维度,就有几维
numpy
code
>>> a = np.arange(6).reshape(2,3)
>>> np.pad(a, ((0,2), (1,1)), "constant")
array([[0, 0, 1, 2, 0],
[0, 3, 4, 5, 0],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0]])
解释