无需广播即可创建大量数组

时间:2019-03-03 22:01:06

标签: python arrays numpy

我有一个数组:

one = np.array([[1], [0]])

如何创建arr使得arr的每个元素都等于one

如果我这样做:

arr = one * 4

由于广播上述陈述并没有达到预期的效果。

1 个答案:

答案 0 :(得分:2)

使用repeat

np.repeat(one[np.newaxis, ...], 4, axis=0)

或者,您可以尝试broadcast_to。这样会更快,但是只读:

np.broadcast_to(one, (4, *one.shape))