python - 无需广播的 Numpy 数组创建

Question

我有一个数组：

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

如何arr创建这样，每个元素arr都等于one？

如果我做：

arr = one * 4

由于广播上述声明没有给出预期的行为。

标签： pythonarraysnumpy

Answer 1

使用repeat：

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

或者，您可以尝试broadcast_to. 这将更快但只读：

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