如何创建一个numpy数组或用另一个数组填充另一个数组？
如何创建一个numpy数组或用另一个数组填充另一个数组？
提问于 2017-04-21 16:17:20
如何创建一个形状为[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回答 4
Stack Overflow用户
回答已采纳
发布于 2017-04-21 16:30:26
Stack Overflow用户
发布于 2017-04-21 16:33:53
如果您的numpy版本是>= 1.10，则可以使用broadcast_to
a = np.arange(1,4)
a.shape = (1,1,3)
b = np.broadcast_to(a,(2,2,3))这会产生一个视图，而不是复制，因此对于大型数组来说会更快。编辑这看起来是您在演示中所要求的结果。
Stack Overflow用户
发布于 2017-04-21 16:34:36
根据Divakar的评论，答案也可以是：
import numpy as np
np.full([2, 2, 3], np.arange(1, 4))另一种可能性是：
import numpy as np
b = np.empty([2, 2, 3])
b[:] = np.arange(1, 4)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/43537439
复制相关文章
相似问题