代码之家  ›  专栏  ›  技术社区  ›  pseudomarvin

矢量化numpy掩码设置

  •  2
  • pseudomarvin  · 技术社区  · 7 年前

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

    如何将以下内容矢量化?

    for i in range(N):
          M[i][y[i]] = values[i]
    
    1 回复  |  直到 7 年前
        1
  •  2
  •   Divakar    7 年前

    简单使用 integer-indexing -

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