我已经看到很多关于如何有效地用特定的东西替换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]