python - 将 numpy 中的矩阵加倍

Question

假设我有一个矩阵 in的size mXn .

我正在尝试创建一个矩阵 out尺寸2mX2n这样out矩阵包含与 in 基本相同的元素矩阵，除了这些值与零交替之外。

例如:

in = [[ 1,2,3],
       [4,5,6]]

out = [[1,0,2,0,3,0],
       [0,0,0,0,0,0],
       [4,0,5,0,6,0],
       [0,0,0,0,0,0]]

有没有矢量化的方法来实现这一点？

Answer 1

使用 NumPy:

import numpy as np

您的数据:

a = np.array([[ 1,2,3],
              [4,5,6]])

创建一个沿两个维度大小都是两倍的数组:

b = np.zeros([x * 2 for x in a.shape], dtype=a.dtype))

将 a 的值分配给 b 的每个第二值，同样在两个维度中:

b[::2,::2] = a

结果:

>>> b
array([[1, 0, 2, 0, 3, 0],
       [0, 0, 0, 0, 0, 0],
       [4, 0, 5, 0, 6, 0],
       [0, 0, 0, 0, 0, 0]])