在 NumPy 数组中快速将元素 i 替换为“[i，i，i]”的 pythonic 方法

Question

我见过很多问题，上面有关于如何有效地用特定的东西替换 NumPy 数组的元素的答案，比如"1"或其他东西，如果它们满足某些条件。

我希望将 2D NumPy 数组中的所有元素替换为自身的数组，即将元素 i 转换为元素[i,i,i]，或者对于我指定的某些函数f(x),g(x),h(x)可能[f(i),g(i),h(i)]。如何以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]