python - 矢量化 numpy 掩码设置

标签 python numpy vectorization array-broadcasting

我有 3 个具有尺寸的 numpy 数组

M (NxC)     # mask
y (N,)     
values (N,)

如何矢量化以下内容?

for i in range(N):
      M[i][y[i]] = values[i]

最佳答案

只需使用 integer-indexing -

M[np.arange(N),y] = values

关于python - 矢量化 numpy 掩码设置,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45170333/

相关文章:

python - Pygame碰撞问题

Python Mechanize.Browser.Open(url) 状态码

python - 高效地将 numpy 数组转换为矩阵

numpy - 用 Scipy 计算两个矩阵的行式点积的矢量化方法

python - 根据另一个整数列表中的重复项更新整数列表的最快方法

python - LinAlg错误: SVD did not converge in matplotlib PCA

python - 处理具有可怕日期时间数据的数据集

python - 使用 pandas 和 numpy 参数化堆栈溢出的用户数和声誉

numpy - 将numpy数组堆栈与自身的厄密转置相乘而无需循环

python - 关于 numpy 沿轴应用和列表理解的困惑