我有一个数组:
one = np.array([[1], [0]])
如何创建arr
使得arr
的每个元素都等于one
?
如果我这样做:
arr = one * 4
由于广播上述陈述并没有达到预期的效果。
答案 0 :(得分:2)
使用repeat
:
np.repeat(one[np.newaxis, ...], 4, axis=0)
或者,您可以尝试broadcast_to
。这样会更快,但是只读:
np.broadcast_to(one, (4, *one.shape))