python - 如何用另一个数组创建或填充一个 numpy 数组？

Question

如何创建形状为 [2, 2, 3] 的 numpy 数组，其中轴 2 的元素是另一个数组，例如 [1, 2, 3]?

所以我想做一些类似这个无效代码的事情:

a = np.arange(1, 4)
b = np.full((3, 3), a)

生成如下数组:

[[[ 1.  2.  3.]
  [ 1.  2.  3.]]
 [[ 1.  2.  3.]
  [ 1.  2.  3.]]]

当然可以使填充循环像这样，但认为可能有一个捷径:

for y in range(b.shape[0]):
    for x in range(b.shape[1]):
        b[y, x, :] = a

Answer 1

有多种方法可以实现这一点。一种是使用 np.full在 np.full((2,2,3), a) 中，正如 Divakar 在评论中指出的那样。或者，您可以使用 np.tile为此，它允许您通过将输入数组重复给定次数来构造一个数组。要构建您的示例，您可以这样做:

import numpy as np

np.tile(np.arange(1, 4), [2, 2, 1])