替换`numpy`数组中的值

Question

我有一个函数，它返回一个numpy整数数组，其值为0到9.我想用10替换所有0。

诀窍是，我不能使用等号，因此我需要一个获取数组并返回一个新数组的函数。因此，这是 NOT 允许：

x = function_returns_numpy_array(...)
x[x == 0] = 10

此外，这是不允许的，因为它会更改数组

x = function_returns_numpy_array(...)
np.place(x, x == 0, 10)

我想要像np.place这样的东西，但实际上会返回新的数组......是否有这样的函数？

WHY ??

因为我想在lambda函数中使用它，例如：

lambda : np.replace_fnc(function_returns_numpy_array(), 0, 10)

因为我是这个人的傻瓜......

Answer 1

您是否正在寻找np.where(x==0,10,x)？