且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

沿轴零填充ndarray

更新时间:2023-02-02 17:07:19

您可以使用 np.pad 并删除最后一行/列:

You can use np.pad and remove the last row/column:

import numpy as np

a = np.array([[1, 2], [3, 4]])
result = np.pad(a, 1, mode='constant', constant_values=0)[:-1, :-1]
print(result)

输出:

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