在python中将零添加到2draay的奇数索引中

Question

我有一个二维数组，我想在数组中的每个元素之间插入零。如下所示：

A = np.array([[1,2,3],[4,5,6],[7,8,9],[10,11,12]])

我想要一些如下的东西：

Result1 = [[1,0,2,0,3],
           [0,0,0,0,0],
           [4,0,5,0,6],
           [0,0,0,0,0],
           [7,0,8,0,9],
           [0,0,0,0,0],
           [10,0,11,0,12]]

谢谢

Answer 1

您可以创建一个新的零数组，并每隔一行/列用切片表示法将A中的值填充到此新数组：

result = np.zeros((2 * A.shape[0] - 1, 2 * A.shape[1] - 1), np.int16)
result[::2, ::2] = A

result
#array([[ 1,  0,  2,  0,  3],
#       [ 0,  0,  0,  0,  0],
#       [ 4,  0,  5,  0,  6],
#       [ 0,  0,  0,  0,  0],
#       [ 7,  0,  8,  0,  9],
#       [ 0,  0,  0,  0,  0],
#       [10,  0, 11,  0, 12]], dtype=int16)