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

Posted 2021-04-02

我已经看到很多关于如何有效地用特定的东西替换NumPy数组元素的答案的问题，如“1”或其他东西，如果它们满足某些条件。

我希望用自己的数组替换2D NumPy数组中的所有元素，即元素i被转换为元素[i,i,i]或者[f(i),g(i),h(i)]，我指定的一些函数f(x),g(x),h(x)。如何以蟒蛇的方式完成这项工作（最好以与Numba相称的方式）？

答案

使用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]