小编Lac*_*ith的帖子

优化numpy ndarray索引操作

我有一个看起来如下的numpy操作:

 for i in range(i_max):
    for j in range(j_max):
        r[i, j, x[i, j], y[i, j]] = c[i, j]
Run Code Online (Sandbox Code Playgroud)

在哪里x,yc具有相同的形状.

是否可以使用numpy的高级索引来加速此操作?

我试过用:

i = numpy.arange(i_max)
j = numpy.arange(j_max)
r[i, j, x, y] = c
Run Code Online (Sandbox Code Playgroud)

但是,我没有得到我预期的结果.

python arrays performance numpy vectorization

5
推荐指数
2
解决办法
107
查看次数

标签 统计

arrays ×1

numpy ×1

performance ×1

python ×1

vectorization ×1