更新除第i个条目以外的numpy数组

Question

我正在尝试实施SGD算法，其中有一个更新公式

使用

可以轻松完成此操作

temp = beta_old[i]
beta = beta_old
beta[i] = temp

但是我发现这很丑陋，我想知道是否还有其他更优雅的方法（也许通过使用一些索引技巧）。

Answer 1

您可能要使用口罩：

mask = np.ones(size, dtype=np.bool)
mask[i] = false

然后再使用面具：

beta[mask] = beta_old[mask]

但是它可能比您当前的方法慢。