python - 如何 "scale"一个 numpy 数组？

Question

我想将形状 (h, w) 的数组缩放 n 倍，得到形状 (h*n, w*n) 的数组。

假设我有一个 2x2 数组:

array([[1, 1],
       [0, 1]])

我想将数组缩放为 4x4:

array([[1, 1, 1, 1],
       [1, 1, 1, 1],
       [0, 0, 1, 1],
       [0, 0, 1, 1]])

即原始数组中每个单元格的值被复制到结果数组中对应的4个单元格中。假设任意数组大小和缩放因子，最有效的方法是什么？

Answer 1

您应该使用 Kronecker product , numpy.kron :

Computes the Kronecker product, a composite array made of blocks of the second array scaled by the first

import numpy as np
a = np.array([[1, 1],
              [0, 1]])
n = 2
np.kron(a, np.ones((n,n)))

给你想要的:

array([[1, 1, 1, 1],
       [1, 1, 1, 1],
       [0, 0, 1, 1],
       [0, 0, 1, 1]])