快速在NumPy数组中用[[i，i，i]`''替换元素i的Python方法？

Question

我已经看到了很多问题，并回答了如何用特定的东西（例如“ 1”或满足某些条件的东西）有效地替换NumPy数组的元素。

我希望用自己的数组替换2D NumPy数组中的所有元素，即对于某些功能{{，元素i变成了元素[i,i,i]或也许是[f(i),g(i),h(i)] 1}}。如何以Python方式做到这一点（最好以一种与Numba一致的方式）？

Answer 1

使用np.repeat：

a = np.repeat(np.arange(30).reshape(10, 3)[..., np.newaxis], 3, axis=2)
print(a.shape)
print(0, 0, :)

输出：

(10, 3, 3)
[0 0 0]